pytorch同一个网络训练,迭代次数不同在cpu上运行速度差异很大
pytorch吧
全部回复
仅看楼主
level 1
synyw5 楼主
pytorch在gpu上训练模型,每迭代5000次保存一次模型,训练完成后发现第5000,10000次的模型在cpu上识别一张图片为1秒,但15000次后的模型识别一张图片要14秒,这是为什么?
2020年03月04日 03点03分 1
level 1
我也遇到了这个问题,老哥解决了吗?
2021年05月16日 09点05分 2
level 1
貌似是Pytorch 1.8的一个bug,Pytorch 1.7就没这个问题,换成1.7吧
2021年07月15日 10点07分 3
1