我训练完成过后，我应该怎么使用？

jdh-algo / JoyHallo

JoyHallo: Digital human model for Mandarin

https://jdh-algo.github.io/JoyHallo/

MIT License

358 stars 35 forks source link

Open wangaocheng opened 1 week ago

wangaocheng commented 1 week ago

按照README的方法，我完成了stage1和stage2的训练，然后我应该如何使用我自己训练后的模型？

DBDXSS commented 1 week ago

您好，可以修改configs/inference/inference.yaml中的audio_ckpt_dir为stage2中您想使用的权重的path

wangaocheng commented 1 week ago

@DBDXSS 感谢你的回复只需要修改这一个参数就可以了对吗？其他不用改了对吗？stage1训练的模型就没用了是吗？

DBDXSS commented 1 week ago

是的，1阶段的模型在2阶段有保留

DBDXSS commented 1 week ago

您可以直接输出net.pth中的权重keys查看是否包含全部需要的权重