2018年12月18日
[PyTorch]论文pytorch复现中遇到的BUG
摘要： [TOC] 1. zip argument 1 must support iteration 在多gpu训练的时候，自动把你的batch_size分成n_gpu份，每个gpu跑一些数据， 最后再合起来。我之所以出现这个bug是因为返回的时候 返回了一个常量。。 2. torch.nn.DataPar 阅读全文
posted @ 2018-12-18 20:43 向前奔跑的少年 阅读(2898) 评论(1) 推荐(0) 编辑