Gluoncv:在gpu上运行train_yolo3.py报ou

2020-03-19  本文已影响0人  AI秘籍
image.png

解决办法:

将gluoncv给的官方训练程序,train_yolo3.py中的batch_size由64改成了1.
这样,程序就可以跑成功了.

image.png

参考:

  1. https://discuss.gluon.ai/t/topic/16788
上一篇 下一篇

猜你喜欢

热点阅读