search

Asad-Ismail / Grounding-Dino-FineTuning

Fine tuning grounding Dino
Apache License 2.0
63 stars 8 forks source link

error: UnpicklingError: invalid load key, 'v'. #3

Closed omer770 closed 7 months ago

omer770 commented 7 months ago

/content/Grounding-Dino-FineTuning/groundingdino/models/GroundingDINO/ms_deform_attn.py:31: UserWarning: Failed to load custom C++ ops. Running on CPU mode Only! warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!") /usr/local/lib/python3.10/dist-packages/torch/functional.py:507: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3549.) return _VF.meshgrid(tensors, kwargs) # type: ignore[attr-defined] final text_encoder_type: bert-base-uncased Traceback (most recent call last): File "/content/Grounding-Dino-FineTuning/train.py", line 11, in model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.py", "weights/groundingdino_swint_ogc.pth") File "/content/Grounding-Dino-FineTuning/groundingdino/util/train.py", line 40, in load_model checkpoint = torch.load(model_checkpoint_path, map_location="cpu") File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1040, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1258, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, 'v'.