search

bubbliiiing / yolo3-pytorch

这是一个yolo3-pytorch的源码，可以用于训练自己的模型。
MIT License
2.01k stars 585 forks source link

是4060显卡跑不了吗？ #190

Open SHuCCai opened 7 months ago

SHuCCai commented 7 months ago

在4060显卡跑的时候，无论怎么降低学习率loss一开始都是nan，当我把train的cuda=true改成cuda=fasle的时候奇迹的能跑了，loss也出值了，但这也太慢了，感觉得跑一天，不能空有4060显卡跑不了吧，不要啊呜呜呜

wh631q commented 4 months ago

因为用得是CPU

wh631q commented 4 months ago

重装环境把，按照步骤一步一步来