文件名称:CDSGD:基于共识的分布式随机梯度下降
文件大小:13KB
文件格式:ZIP
更新时间:2024-05-30 23:35:27
Python
CDSGD 这是基于共识的分布式随机梯度下降的占位符存储库。 有关更多详细信息,请参阅论文:姜占宏,阿迪亚·巴鲁,金美·黑德,苏米克·萨卡 用法 python main.py -m CNN -b 512 -ep 200 -d cifar10 -n 5 -cp 1 -g 3 CDSGD -m是模型名称,它是CNN,FCN和Big_CNN; -b批处理大小; -ep是时代; -d是数据集; -n是否。 代理商; -cp是通信周期; -g是您要使用的GPU。 假设您有4个GPU,然后选择要使用的GPU。 然后终于要进行的实验了。 SGD,CDSGD,EASGD,CDMSGD,MSGD,FASGD等 执照 BSD
【文件预览】:
CDSGD-master
----FASGD.py(1KB)
----train.py(16KB)
----EASGD.py(4KB)
----main.py(2KB)
----models.py(3KB)
----LICENSE(1KB)
----eval.py(2KB)
----README.md(781B)
----CDSGD.py(4KB)
----CDMSGD.py(4KB)
----.gitignore(1KB)