-
Notifications
You must be signed in to change notification settings - Fork 13
Open
Description
Is this code also support CPU version of mc module. If yes, which one it would be?
I am having the following error
File "models/export.py", line 33, in <module>
model = attempt_load(opt.weights, map_location=torch.device('cpu')) # load FP32 model
File "./models/experimental.py", line 118, in attempt_load
model.append(torch.load(w, map_location=map_location)['model'].float().fuse().eval()) # load FP32 model
File "/home/rokeshar/.local/lib/python3.7/site-packages/torch/serialization.py", line 608, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/rokeshar/.local/lib/python3.7/site-packages/torch/serialization.py", line 787, in _legacy_load
result = unpickler.load()
ModuleNotFoundError: No module named 'mc'
Metadata
Metadata
Assignees
Labels
No labels