[Python] Why do I get such an error when I add the ConvNeXtV2 module code to YOLOv8?

Discussão em 'Python' iniciado por Stack, Abril 28, 2025.

  1. Stack

    Stack Membro Participativo

    Traceback (most recent call last):
    File "D:\Personal\Desktop\yolov8\demo-train.py", line 8, in <module>
    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.

    Continue reading...

Compartilhe esta Página