search

river-huang / ARWGAN

This repository is the official PyTorch implementation of ARWGAN: attention-guided robust image watermarking model based on GAN.
MIT License
28 stars 6 forks source link

作者您好，我想咨询一下有关模型训练的相关问题。 #17

Open udmekey opened 3 months ago

udmekey commented 3 months ago

作者您好，我想咨询一下您。请问模型训练batchsize为多少合适呢、epoch=100是否合适、我尝试batchsize=32时，模型在epoch=7后loss在下降到0.55左右，后续模型训练就不再收敛的原因。 [Uploading 问题描述.docx…]()

river-huang commented 3 months ago

python main.py new --name 'ARWGAN' --data-dir /media/dell/PROG/HJT/data/coco/ --batch-size 16 --epoch 50 --noise "crop((0.1871,0.1871),(0.1871,0.1871))+Jpeg(0)+dropout(0.5477,0.5477)+resize(0.8,0.8)+gaussian(3,2.0)" 你可以尝试下这种设置，这是我当时训练时的设置。如果依旧无法收敛，可以查看是否模型结构或者其他代码发生改变。我们提供了完整的预训练模型可以作为参考

udmekey commented 3 months ago

感谢作者能够这么详细解答我的疑问。在数字水印方向开源代码并不多，作者不但提供了新的解决方案还将代码开源共享，并在代码中做了详细标注。由衷地感谢您，期待您后续的分享。