Closed wwq66 closed 7 months ago
Hi, what's your torch version?
You can try pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
I've tried torch version of 1.12.1, 1.13.1, 2.1.0, but none of these version works. And the pytorch docs shows that torch.load do not have the keyword 'strict' as below.
Only the load_state_dict have the 'strict' keyword
I see. I think you can directly remove the argument of strict=False
. I will fix it later.
Removing strict
can actually fix the error. But seems to raise another error about missing keys while loading state_dict. I'll close this issue temporarily and try to fix the missing keys error.
raise TypeError: 'strict' is an invalid keyword argument for Unpickler() when running the code below. It seems torch.load does not have the keyword 'strict'.