Open zgeh994 opened 11 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主。