search

alibaba / FederatedScope

An easy-to-use federated learning platform
https://www.federatedscope.io
Apache License 2.0
1.26k stars 206 forks source link

Pytorch model saving #741

Closed kamelferrahi closed 7 months ago

kamelferrahi commented 8 months ago

HI, I need clarification on saving a model trained in distributed mode with a .pth file extension on how to do it and where to find the saved model.

rayrayraykk commented 8 months ago

Use federate.save_to to save the model. And you can see full configs in https://github.com/alibaba/FederatedScope/tree/master/federatedscope/core/configs.