[PyTorch]论文pytorch复现中遇到的BUG
Posted kk17
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了[PyTorch]论文pytorch复现中遇到的BUG相关的知识，希望对你有一定的参考价值。
目录
1. zip argument #1 must support iteration
在多gpu训练的时候，自动把你的batch_size分成n_gpu份，每个gpu跑一些数据， 最后再合起来。我之所以出现这个bug是因为返回的时候 返回了一个常量。。
2. torch.nn.DataParallel
在使用torch.nn.DataParallel时候，要先把模型放在gpu上，再进行parallel。
以上是关于[PyTorch]论文pytorch复现中遇到的BUG的主要内容，如果未能解决你的问题，请参考以下文章