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