john-zhaofei
commented
4 days ago Maybe the pre-trained model was not downloaded correctly.
Open simplcup opened 5 days ago
john-zhaofei
commented
4 days ago Maybe the pre-trained model was not downloaded correctly.
m C:\Margio\SepReformer-main\models\SepReformer_Base_WSJ0\log\scratch_weights\epoch.0180.pth ..... Traceback (most recent call last): File "C:\Margio\SepReformer-main\run.py", line 32, in
main_module.main(args)
File "C:\Margio\SepReformer-main\utils\decorators.py", line 12, in wrapped
result = func(*args, *kwargs)
File "C:\Margio\SepReformer-main\models\SepReformer_Base_WSJ0\main.py", line 43, in main
engine = Engine(args, config, model, dataloaders, criterions, optimizers, schedulers, gpuid, device)
File "C:\Margio\SepReformer-main\utils\decorators.py", line 12, in wrapped
result = func(args, kwargs)
File "C:\Margio\SepReformer-main\models\SepReformer_Base_WSJ0\engine.py", line 36, in init
self.start_epoch = util_engine.load_last_checkpoint_n_get_epoch(self.checkpoint_path, self.model, self.main_optimizer, location=self.device)
File "C:\Margio\SepReformer-main\utils\util_engine.py", line 42, in load_last_checkpoint_n_get_epoch
checkpoint_dict = torch.load(latest_checkpoint_file, map_location=location)
File "C:\Users\atom4\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\serialization.py", line 1384, in load
return _legacy_load(
File "C:\Users\atom4\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\serialization.py", line 1628, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.
i get this error when trying to use the model, can someone please help me?