使用PyTorch搭建神經網絡在訓練過程中,可能遇見的一些bug。

bug-01:THTensor.cpp:689 和 THTensor.cpp:711

原因如下:

  1. 使用Dataloader加載的圖片,使用參數batch_size大於1,一個batch_size的圖片集中彼此圖片大小不一致。
  2. 圖片集的數量不能整除batch_size。
bug-02:copy_if failed to synchronize: device-side assert triggered

原因:這是由於加載的GroundTruth數據標籤不是從0開始或超過num_classes導致。

bug-03 cuda runtime error (59)

原因:同上。

參考
  • https://www.cnblogs.com/zxj9487/p/11531888.html
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章