Open zgeh994 opened 8 months ago
老师,我电脑cpu训练,然后运行train.py时候训练的很慢,python3.6的,torch1.2的,torch vision0.4的,训练模型一天一夜都没运行完事,然后我运行其他的predict.py显示cuda错误, 1、是没有训练出来结果就不能进行下一步是吗,我电脑很慢,能不能给你训练好的模型发我一下,我跟你环境差不多,代码也没改,就改了你说的那部分, 2、我用的cpu,怎么会有cuda报错呢 3、import onnxsim怎么安装,还是不耽误? 我是看了你的b站视频,修改了voc_annotation.py里面的annotation_mode=2,生成2007_train.txt和2007_val.txt。然后train.py训练一天一夜我就暂停了。之后yolo.py里面model_path以及classes_path, "model_path" : 'D:/opencv-4.7.0-windows/shenjieke/logs/best_epoch_weights.pth', "classes_path" : 'D:/opencv-4.7.0-windows/shenjieke/model_data/new_classes.txt', best_epoch_weights.pth是我自己瞎放的,因为模型没训练出来 在logs文件夹里目前有best_epoch_weights.pth和last_epoch_weights.pth 目前运行predict.py也没有框。 归根结底是不是就是模型没有训练出来所以下一步不好使?那我也没用cuda呀,谢谢up主。
老师,我电脑cpu训练,然后运行train.py时候训练的很慢,python3.6的,torch1.2的,torch vision0.4的,训练模型一天一夜都没运行完事,然后我运行其他的predict.py显示cuda错误, 1、是没有训练出来结果就不能进行下一步是吗,我电脑很慢,能不能给你训练好的模型发我一下,我跟你环境差不多,代码也没改,就改了你说的那部分, 2、我用的cpu,怎么会有cuda报错呢 3、import onnxsim怎么安装,还是不耽误? 我是看了你的b站视频,修改了voc_annotation.py里面的annotation_mode=2,生成2007_train.txt和2007_val.txt。然后train.py训练一天一夜我就暂停了。之后yolo.py里面model_path以及classes_path, "model_path" : 'D:/opencv-4.7.0-windows/shenjieke/logs/best_epoch_weights.pth', "classes_path" : 'D:/opencv-4.7.0-windows/shenjieke/model_data/new_classes.txt', best_epoch_weights.pth是我自己瞎放的,因为模型没训练出来 在logs文件夹里目前有best_epoch_weights.pth和last_epoch_weights.pth 目前运行predict.py也没有框。 归根结底是不是就是模型没有训练出来所以下一步不好使?那我也没用cuda呀,谢谢up主。