search

Royalvice / DocDiff

ACM Multimedia 2023: DocDiff: Document Enhancement via Residual Diffusion Models. Also contains 1597 red seals in Chinese scenes, along with their corresponding binary masks.
https://www.aibupt.com/
MIT License
196 stars 21 forks source link

训练集规模以及学习率 #15

Closed yrqs closed 9 months ago

yrqs commented 9 months ago

作者您好，我想问一下训练去印章的时候数据集的规模大概是多少呢？以及学习率和batchsize相关吗？我如果调小batchsize的话，学习率是否需要在conf.yml中手动缩放？谢谢！

Royalvice commented 9 months ago

作者您好，我想问一下训练去印章的时候数据集的规模大概是多少呢？以及学习率和batchsize相关吗？我如果调小batchsize的话，学习率是否需要在conf.yml中手动缩放？谢谢！

我们最终使用了差不多8000样本量进行训练。学习率和batchsize相关性，我并未做较深研究。以我的实验来看，相关性不强。以训练集中间的可视化和loss结果和验证集结果为辅助判断收敛效果

yrqs commented 9 months ago

好的，十分感谢！