Traceback (most recent call last):
File "D:\Personal\Desktop\yolov8\demo-train.py", line 8, in
model = YOLO("ultralytics/cfg/models/v8/ConvNextv2_yolov8.yaml")
File "D:\Personal\Desktop\yolov8\ultralytics\models\yolo\model.py", line 23, in __init__
super().__init__(model=model, task=task, verbose=verbose)
File "D:\Personal\Desktop\yolov8\ultralytics\engine\model.py", line 143, in __init__
self._new(model, task=task, verbose=verbose)
File "D:\Personal\Desktop\yolov8\ultralytics\engine\model.py", line 251, in _new
self.model = (model or self._smart_load("model"))(cfg_dict, verbose=verbose and RANK == -1) # build model
File "D:\Personal\Desktop\yolov8\ultralytics\nn\tasks.py", line 379, in __init__
self.model, self.save = parse_model(deepcopy(self.yaml), ch=ch, verbose=verbose) # model, savelist
File "D:\Personal\Desktop\yolov8\ultralytics\nn\tasks.py", line 1107, in parse_model
m = m(*args)
File "D:\Personal\Desktop\yolov8\ultralytics\nn\modules\ConvNextv2.py", line 167, in convnextv2_atto
model.load_state_dict(update_weight(model.state_dict(), torch.load(weights)['model']))
File "D:\Personal\Desktop\yolov8\ultralytics\utils\patches.py", line 86, in torch_load
return _torch_load(*args, **kwargs)
File "D:\YushuaiA\envs\v8\lib\site-packages\torch\serialization.py", line 594, in load
with _open_file_like(f, 'rb') as opened_file:
File "D:\YushuaiA\envs\v8\lib\site-packages\torch\serialization.py", line 235, in _open_file_like
return _open_buffer_reader(name_or_buffer)
File "D:\YushuaiA\envs\v8\lib\site-packages\torch\serialization.py", line 220, in __init__
_check_seekable(buffer)
File "D:\YushuaiA\envs\v8\lib\site-packages\torch\serialization.py", line 311, in _check_seekable
raise_err_msg(["seek", "tell"], e)
File "D:\YushuaiA\envs\v8\lib\site-packages\torch\serialization.py", line 304, in raise_err_msg
raise type(e)(msg)
AttributeError: 'int' object has no attribute 'seek'. You can only torch.load from a file that is seekable. Please pre-load the data into a buffer like io.BytesIO and try to load from it instead.
Подробнее здесь: https://stackoverflow.com/questions/795 ... -to-yolov8
Почему я получаю такую ошибку, когда добавляю код модуля ConvnextV2 в Yolov8? ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение