search

apple / pfl-research

Simulation framework for accelerating research in Private Federated Learning
http://apple.github.io/pfl-research/
Apache License 2.0
285 stars 27 forks source link

dtype argument in to_tensor #24

Closed grananqvist closed 7 months ago

grananqvist commented 7 months ago

torch was float32 always, tensorflow was none always. parametrize this and default to float32.