使用PyTorch搭建神经网络在训练过程中,可能遇见的一些bug。
bug:THTensor.cpp:689 和 THTensor.cpp:711
原因如下:
使用Dataloader加载的图片,使用参数batch_size大于1,一个batch_size的图片集中彼此图片大小不一致。
图片集的数量不能整除batch_size.
参考
https://www.cnblogs.com/zxj9487/p/11531888.html
...
原创
2019-11-17 15:30:53 ·
360 阅读 ·
0 评论