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

fix shared cuda tensor issue #19

Closed Xiao-Chenguang closed 1 month ago

Xiao-Chenguang commented 1 month ago

del the tensor from subprocess. Ensure the tensor produced by the subprocess is cleared before closing the subprocess.