文件名称：DDG:PyTorch实施“具有收敛保证的解耦并行反向传播”
文件大小：234KB
文件格式：ZIP
更新时间：2024-05-20 13:09:58
Python
介绍 该存储库是的论文的PyTorch代码 如有任何疑问或建议，请随时联系 依存关系 Python 3.6.2（Anaconda） PyTorch 0.3.1 CUDA 8.0 跑步 克隆并cd进入repo目录，运行： python main.py ./data/cifar.python --dataset cifar10 --arch resnet_ddg --epochs 300 --learning_rate 0.01 --schedule 150 225 --gammas 0.1 0.1 --batch_size 128 --manualSeed 2 --depth 110 --splits 2 结果 使用DDG优化CIFAR10上ResNet110的实验结果。 请注意，要获得本文所述的加速效果，代码应在具有4个GPU的计算机上运行。 引文 @article{huo20
【文件预览】：
DDG-master
----models()
--------__pycache__()
--------__init__.py(54B)
--------resnet_ddg.py(6KB)
----main.py(14KB)
----utils.py(2KB)
----__pycache__()
--------sharedtensor.cpython-36.pyc(2KB)
--------utils.cpython-36.pyc(3KB)
----results()
--------result.jpg(242KB)
----README.md(1KB)
----sharedtensor.py(2KB)