BasicSR icon indicating copy to clipboard operation
BasicSR copied to clipboard
verified publisher icon XPixelGroup

【BUG】模型提前终止训练，进入latest model的推理

Open 11923303233 opened this issue 2 years ago • 3 comments

我使用844张图片生成lmdb训练，设置50W iter，但是在490600时保存latest model模型，进入最后的验证阶段。

11923303233 avatar Sep 19 '23 02:09 11923303233

你好，我也遇到了和你相同的问题，训练提前终止，请问你找到原因了吗。

guoxiangyv avatar Oct 17 '23 07:10 guoxiangyv

同问，不知道为什么训练提前终止

JianHaoYin avatar Mar 11 '24 14:03 JianHaoYin

我使用844张图片生成lmdb训练，设置50W iter，但是在490600时保存latest model模型，进入最后的验证阶段。

它的停止逻辑有两个，iter和epoch，有一个达到阈值就会停止，可以将dataset_enlarge_ratio设置得大一点

Bruce-WangGF avatar Jun 19 '24 14:06 Bruce-WangGF