使用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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章