-
Notifications
You must be signed in to change notification settings - Fork 132
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
yolov5哪个版本 #15
Comments
通过自己的模型,使用V3.0版本成功跑通,jetson nano 10帧 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
这个pt模型是用哪个版本训练的,我试了v5.0和v3.0都不行。V5可以转wts,但是engine出问题,V3都转不了wts
Traceback (most recent call last):
File "gen_wts.py", line 8, in
model = torch.load('weights/number_v5s.pt', map_location=device)['model'].float() # load to FP32
File "/home/lpx/.local/lib/python3.8/site-packages/torch/serialization.py", line 789, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File "/home/lpx/.local/lib/python3.8/site-packages/torch/serialization.py", line 1131, in _load
result = unpickler.load()
File "/home/lpx/.local/lib/python3.8/site-packages/torch/serialization.py", line 1124, in find_class
return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'C3' on <module 'models.common' from '/home/lpx/ros/yolov5v3.0/src/yolov5/models/common.py'>
The text was updated successfully, but these errors were encountered: