Load weights model_data/yolov4_tiny_weights_coco.pth.
Traceback (most recent call last):
File "C:/Users/刘/Desktop/yolov4-tiny-pytorch/train.py", line 225, in
pretrained_dict = torch.load(model_path, map_location = device)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 386, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 573, in _load
result = unpickler.load()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 536, in persistent_load
deserialized_objects[root_key] = restore_location(obj, location)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 406, in restore_location
return default_restore_location(storage, str(map_location))
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 119, in default_restore_location
result = fn(storage, location)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 98, in _cuda_deserialize
with torch.cuda.device(device):
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda__init.py", line 243, in enter
self.prev_idx = torch._C._cuda_getDevice()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda__init__.py", line 180, in _lazy_init
_cudart = _load_cudart()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda__init__.py", line 77, in _load_cudart
lib = find_cuda_windows_lib()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda\init__.py", line 50, in find_cuda_windows_lib
out = out.decode().strip()
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xc1 in position 9: invalid start byte
bubbliiiing
commented
2 years ago
Load weights model_data/yolov4_tiny_weights_coco.pth. Traceback (most recent call last): File "C:/Users/刘/Desktop/yolov4-tiny-pytorch/train.py", line 225, in
pretrained_dict = torch.load(model_path, map_location = device)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 386, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 573, in _load
result = unpickler.load()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 536, in persistent_load
deserialized_objects[root_key] = restore_location(obj, location)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 406, in restore_location
return default_restore_location(storage, str(map_location))
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 119, in default_restore_location
result = fn(storage, location)
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\serialization.py", line 98, in _cuda_deserialize
with torch.cuda.device(device):
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda__init.py", line 243, in enter
self.prev_idx = torch._C._cuda_getDevice()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda__init__.py", line 180, in _lazy_init
_cudart = _load_cudart()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda__init__.py", line 77, in _load_cudart
lib = find_cuda_windows_lib()
File "C:\Users\刘\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\cuda\init__.py", line 50, in find_cuda_windows_lib
out = out.decode().strip()
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xc1 in position 9: invalid start byte
请问要怎么解决