Pytorch model saving
kamelferrahi opened this issue · 1 comments
kamelferrahi commented
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
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.