摘要:
[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
向前奔跑的少年
阅读(2931)
评论(1)
推荐(0)
浙公网安备 33010602011771号