-
Notifications
You must be signed in to change notification settings - Fork 17
Description
Hi, I've run Train.py successfully and the Test.py script works up until line 285, where I get the following:
Traceback (most recent call last): File "Test.py", line 362, in <module> test(test_dataframe) File "Test.py", line 285, in test model.load_state_dict(torch.load(Model_Path + model_name,map_location='cpu')) File "/home/oroscow/micromamba/envs/bioinfo/lib/python3.8/site-packages/torch/serialization.py", line 815, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/home/oroscow/micromamba/envs/bioinfo/lib/python3.8/site-packages/torch/serialization.py", line 1033, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '\x0a'.
I've read online that this has to do with either the protocol version or encoding style being used by PyTorch but I'm not familiar enough with the scripts or the package to know how to resolve this. Thanks for any and all help :)