Thank you for your excellent work. I noticed that the size of the pre-trained model is only 259KB when I tried to use it, and I encountered an error while loading it. Could there be an issue with the size of this model?
Total number of parameters: 18.366555 M
Traceback (most recent call last):
File "eval_4x.py", line 74, in
model.load_state_dict(torch.load(model_name))
File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 608, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 777, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
_pickle.UnpicklingError: invalid load key, '\x0a'.
Thank you for your excellent work. I noticed that the size of the pre-trained model is only 259KB when I tried to use it, and I encountered an error while loading it. Could there be an issue with the size of this model?
Total number of parameters: 18.366555 M Traceback (most recent call last): File "eval_4x.py", line 74, in
model.load_state_dict(torch.load(model_name))
File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 608, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 777, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
_pickle.UnpicklingError: invalid load key, '\x0a'.