update serialization tools

SMILELab-FL / FedLab

A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.

https://fedlab.readthedocs.io

Apache License 2.0

740 stars 127 forks source link

Closed AgentDS closed 1 year ago

AgentDS commented 1 year ago

update serialization tools for subtraction parameters, for both trainable/untrainable params;
add 'cpu' option for serialization functions;
update docstring for serialization tools