pytorch RuntimeError: Expected to have finished reduction in the prior iteration before starting a n

最新推荐文章于 2024-07-13 23:34:27 发布

城俊BLOG

最新推荐文章于 2024-07-13 23:34:27 发布

阅读量6.6k

点赞数 7

分类专栏： Pytorch # DL-报错

本文链接：https://blog.csdn.net/qxqxqzzz/article/details/116076355

版权

DL-报错同时被 2 个专栏收录

122 篇文章 11 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

Pytorch

47 篇文章 4 订阅

订阅专栏

本文介绍了在使用PyTorch时遇到的RuntimeError：Expected to have finished reduction错误的三种解决方法。包括调整DistributedDataParallel参数、避免计算图外露以及在网络有多个输入时正确处理前向和反向传播的步骤。通过这些解决方案，可以成功避免类似错误的发生。

摘要由CSDN通过智能技术生成

报错：

Traceback (most recent call last):
  File "train.py", line 166, in <module>
    main(args_)
  File "train.py", line 118, in main
    f_clean_masked, f_occ_masked, fc, fc_occ = backbone(img1, img2)
  File "