-
Notifications
You must be signed in to change notification settings - Fork 32
Description
The error message is as follows, really appreciate your help.
Traceback (most recent call last):
File "D:/NEU-DET-Steel-Surface-Defect-Detection-master/FPN_Resnet_Train.py", line 31, in
model_trainer = Trainer(model, lr, epochs, path)
File "D:\NEU-DET-Steel-Surface-Defect-Detection-master\Utilities\Trainer.py", line 54, in init
checkpoint = torch.load(self.PATH)
File "C:\Users\pc.conda\envs\env\lib\site-packages\torch\serialization.py", line 593, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "C:\Users\pc.conda\envs\env\lib\site-packages\torch\serialization.py", line 762, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.
Process finished with exit code 1