DiffSynth-Studio icon indicating copy to clipboard operation
DiffSynth-Studio copied to clipboard

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

Open luoan7248 opened this issue 1 year ago • 1 comments

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

luoan7248 avatar Jul 26 '24 03:07 luoan7248

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

Artiprocher avatar Jul 29 '24 05:07 Artiprocher