微调加载模型报错

[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'