1. 程式人生 > >[PyTorch]論文pytorch復現中遇到的BUG

[PyTorch]論文pytorch復現中遇到的BUG

apar href ati 訓練 論文 size 出現 數據 使用

目錄

  • 1. zip argument #1 must support iteration
  • 2. torch.nn.DataParallel

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