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
2023-01-24 11:33:17
13KB
Python
1