Add to_tensor in PyTorchModel

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

Closed congzheng-song closed 8 months ago

congzheng-song commented 8 months ago

This enables passing numpy array to PyTorchModel.do_multiple_epochs_of.