modelscope / DiffSynth-Studio

Enjoy the magic of Diffusion models!
Apache License 2.0
6.59k stars 600 forks source link

可图微调的权重是多个,如何确定最佳的微调权重呢?请问如何获取loss图? #125

Open luoan7248 opened 3 months ago

luoan7248 commented 3 months ago

image 请问哪一个是最佳的权重,如何通过查看loss值或者图查看查看训练的效果好不好呢?

Artiprocher commented 3 months ago

loss 曲线可以通过启动 tensorboard 来实时查看,对于 vscode 用户,安装 tensorboard 插件后,按下 shift + command + p,选择 Python: Launch Tensorboard 即可。但由于 diffusion 模型的特殊性,loss 并不代表实际的生成效果,当 loss 很小时,模型可能已经完全过拟合,只能生成训练的那几张图,因此我建议用每个权重文件都测试一下,手动选择效果最好的版本。