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



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

  

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

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

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




  

相关话题

  硕士方向,选择迁移学习还是自然语言处理? 
  如何评价Google最新提出的gMLP:MLP模型在CV和NLP任务上均取得较好的效果? 
  请问下大家训练 SimCSE 时, loss 有没有这样的情况? 
  ICML2020有哪些值得关注的工作? 
  学生网络用知识蒸馏损失去逼近教师网络,如何提高学生网络的准确率? 
  深度学习图像处理什么时候用到GPU? 
  卷积神经网络中卷积核是如何学习到特征的? 
  搞机器学习的生环化材是天坑吗? 
  目前 AI 在疾病的诊断和治疗上,有哪些成功的应用? 
  新手如何入门pytorch? 

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





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