NEU-DET-Steel-Surface-Defect-Detection icon indicating copy to clipboard operation
NEU-DET-Steel-Surface-Defect-Detection copied to clipboard

I ran the script and got an error. How can I fix it?

Open Shiratsuyu1 opened this issue 2 years ago • 1 comments

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

Shiratsuyu1 avatar Dec 01 '21 03:12 Shiratsuyu1