search

hw-du / CBiT

Implementation of the paper "Contrastive Learning with Bidirectional Transformers for Sequential Recommendation".
GNU General Public License v3.0
26 stars 6 forks source link

when test model #2

Closed arnold-em closed 1 year ago

arnold-em commented 1 year ago

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