微调加载模型报错

[Ranzige]

作者好，我在微调模型的时候会报这种错误：Restoring states from the checkpoint path at /data/xuruan/guiji_projects/SingingVocoders-main/exp/laoban_hifi/model_ckpt_steps_0.ckpt
Traceback (most recent call last):
File "train.py", line 111, in
train()
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/click/core.py", line 1157, in call
return self.main(*args, **kwargs)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/click/core.py", line 1078, in main
rv = self.invoke(ctx)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/click/core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/click/core.py", line 783, in invoke
return __callback(*args, **kwargs)
File "train.py", line 104, in train
trainer.fit(task, ckpt_path=get_latest_checkpoint_path(work_dir))
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/trainer/trainer.py", line 543, in fit
call._call_and_handle_interrupt(
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/trainer/call.py", line 44, in _call_and_handle_interrupt
return trainer_fn(*args, **kwargs)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/trainer/trainer.py", line 579, in _fit_impl
self._run(model, ckpt_path=ckpt_path)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/trainer/trainer.py", line 955, in _run
self._checkpoint_connector._restore_modules_and_callbacks(ckpt_path)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/trainer/connectors/checkpoint_connector.py", line 397, in _restore_modules_and_callbacks
self.resume_start(checkpoint_path)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/trainer/connectors/checkpoint_connector.py", line 80, in resume_start
self._loaded_checkpoint = _pl_migrate_checkpoint(loaded_checkpoint, checkpoint_path)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/utilities/migration/utils.py", line 143, in _pl_migrate_checkpoint
old_version = _get_version(checkpoint)
File "/data/xuruan/anaconda3/envs/reflow-svc/lib/python3.8/site-packages/lightning/pytorch/utilities/migration/utils.py", line 164, in _get_version
return checkpoint["pytorch-lightning_version"]
KeyError: 'pytorch-lightning_version'