百科问答小站 logo
百科问答小站 font logo



pytorch 分布式计算 你们都遇到过哪些 坑/bug? 第1页

  

user avatar   fan-ru-chao 网友的相关建议: 
      

说下我之前遇到的坑,如果是用pytorch实现同步梯度更新,然后数据接口是自己写的话一定要注意保证每张卡分配的batch数是一样的。因为如果某张卡少了一个batch的话,其他卡就会等待,从而程序卡在torch.all_reduce()上。最后的情况就会出现在第一个epoch结尾处程序卡住,而且没有报错信息。

当时调了半天,一度以为是pytorch的某种bug,最后检查数据接口才发现了这个小错误。




  

相关话题

  2022 年初,你认为哪项成果代表了现在人工智能的最高水平? 
  为什么yolov5从零开始训练(不在ImageNet上预训练)能够达到如此高的性能? 
  attention跟一维卷积的区别是啥? 
  如何评价「Patches are all you need」? 
  神经网络训练多个epoch,写论文的时候可以取最好的效果那一个epoch作为结果吗? 
  如何看待 AlphaFold 在蛋白质结构预测领域的成功? 
  Google 的神经网络生成图像 (Inceptionism) 是怎么做到的? 
  百度学术能否替代 Google 学术网站?有没有其他选择? 
  普通FPGA工程师怎样向人工智能靠拢? 
  全连接层的作用是什么? 

前一个讨论
怎样理解阻塞非阻塞与同步异步的区别?
下一个讨论
国内做NLP业务的香侬科技值得去吗?





© 2025-03-04 - tinynew.org. All Rights Reserved.
© 2025-03-04 - tinynew.org. 保留所有权利