Arcface系列(五) Pytorch版本的单机多卡分布式训练

Arcface系列(五) Pytorch版本的单机多卡分布式训练_第1张图片

折腾了一天终于完成了。可以继续摸鱼了,开熏~~~~~~(吐槽大会王建国语气~)

分布式训练参考:

单卡pytorch 版本的Arcface代码参考:俺也是在这上面修改的,谢谢这位大哥的repo

https://github.com/TreB1eN/InsightFace_Pytorch

当代研究生应当掌握的并行训练方法(单机多卡)

https://zhuanlan.zhihu.com/p/98535650?d=1577237968168

PyTorch分布式训练

https://www.jianshu.com/p/ed47a6499dae

一个BUG

https://github.com/pytorch/pytorch/issues/8637

你可能感兴趣的:(...)