标签：训练 class zip ati must 遇到 data ref support
目录
在多gpu训练的时候，自动把你的batch_size分成n_gpu份，每个gpu跑一些数据， 最后再合起来。我之所以出现这个bug是因为返回的时候 返回了一个常量。。
在使用torch.nn.DataParallel时候，要先把模型放在gpu上，再进行parallel。
标签：训练 class zip ati must 遇到 data ref support
原文地址：https://www.cnblogs.com/kk17/p/10139884.html
