MuseTalk icon indicating copy to clipboard operation
MuseTalk copied to clipboard

The saved training model is pth, while resume_from_checkpoint: True , How to solve the problem of loading bin?

Open sunbo11112 opened this issue 8 months ago • 3 comments

The saved training model is pth, while resume_from_checkpoint: True , How to solve the problem of loading bin?

sunbo11112 avatar Apr 17 '25 04:04 sunbo11112

Hi @sunbo11112, this line of code will save the checkpoint in .bin format. If you set resume_from_checkpoint toTrue, the code will automatically locate the latest checkpoint and resume training from there.

zzzweakman avatar Apr 17 '25 09:04 zzzweakman

嗨,这代码将以格式保存检查点。如果设置为,代码将自动找到最新的检查点并从那里恢复训练。.bin``resume_from_checkpoint``True

save_model_epoch_interval=1 Training for one epoch, the code will save the model in bin format, right @zzzweakman

sunbo11112 avatar Apr 17 '25 09:04 sunbo11112