Reproductivity problem with cuda

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

Closed Xiao-Chenguang closed 4 weeks ago

Xiao-Chenguang commented 4 weeks ago

Setting SEED does make the experiments reproducible under CPU execution, but not with CUDA.

Xiao-Chenguang commented 4 weeks ago

The example fedavg_demo.py is reproducible (with CPU). cuda_demo.py is not (with cuda). PyTorch Reproducibility may helps.