-
Notifications
You must be signed in to change notification settings - Fork 443
Description
when i resume from instant_mesh_base.ckpt,using "python train.py --resume /ckpts/instant_mesh_base.ckpt --resume_weights_only --base configs/instant-mesh-base-train-finetune.yaml --gpus 0,1 --num_nodes 1".
ERROR:
Traceback (most recent call last):
File "/model/2023/XXX/InstantMesh-main/train.py", line 186, in
model = model.class.load_from_checkpoint(opt.resume, **config.model.params)
File "/home/admin/anaconda3/envs/mesh/lib/python3.10/site-packages/pytorch_lightning/core/module.py", line 1552, in load_from_checkpoint
loaded = _load_from_checkpoint(
File "/home/admin/anaconda3/envs/mesh/lib/python3.10/site-packages/pytorch_lightning/core/saving.py", line 64, in _load_from_checkpoint
checkpoint = _pl_migrate_checkpoint(
File "/home/admin/anaconda3/envs/mesh/lib/python3.10/site-packages/pytorch_lightning/utilities/migration/utils.py", line 142, in _pl_migrate_checkpoint
old_version = _get_version(checkpoint)
File "/home/admin/anaconda3/envs/mesh/lib/python3.10/site-packages/pytorch_lightning/utilities/migration/utils.py", line 165, in _get_version
return checkpoint["pytorch-lightning_version"]
KeyError: 'pytorch-lightning_version'