Xiao-Chenguang / FedMind

Federated Learning framework for researcher on top of PyTorch.
https://github.com/Xiao-Chenguang/FedMind
MIT License
1 stars 0 forks source link

Support StateDict inplace operations #64

Closed Xiao-Chenguang closed 1 week ago

Xiao-Chenguang commented 1 week ago

Add inplace math operations +, -, *, and / to StateDict class. This keeps the reference and will be useful if the references are used.