Traceback (most recent call last):
File "D:\download\CBiT-master\CBiT-master\test.py", line 23, in
train()
File "D:\download\CBiT-master\CBiT-master\test.py", line 18, in train
trainer.test()
File "D:\download\CBiT-master\CBiT-master\trainers\base.py", line 228, in test
best_model = torch.load(os.path.join(self.export_root, 'models', 'best_acc_model.pth')).get('model_state_dict')
File "D:\ProgramData\Anaconda3\envs\cbit\lib\site-packages\torch\serialization.py", line 608, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "D:\ProgramData\Anaconda3\envs\cbit\lib\site-packages\torch\serialization.py", line 777, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.
Test best model with test set!
is there wrong?
thank
Traceback (most recent call last): File "D:\download\CBiT-master\CBiT-master\test.py", line 23, in
train()
File "D:\download\CBiT-master\CBiT-master\test.py", line 18, in train
trainer.test()
File "D:\download\CBiT-master\CBiT-master\trainers\base.py", line 228, in test
best_model = torch.load(os.path.join(self.export_root, 'models', 'best_acc_model.pth')).get('model_state_dict')
File "D:\ProgramData\Anaconda3\envs\cbit\lib\site-packages\torch\serialization.py", line 608, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "D:\ProgramData\Anaconda3\envs\cbit\lib\site-packages\torch\serialization.py", line 777, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.
Test best model with test set!
is there wrong?
thank