search

ChrisChen1023 / HINT

HINT: High-quality INpainting Transformer with Enhanced Attention and Mask-aware Encoding
MIT License
30 stars 4 forks source link

训练过程中是否保存的是最优的模型文件 #19

Closed Idol-Dou2021 closed 4 months ago

Idol-Dou2021 commented 4 months ago

您好，我们非常认可您的工作。请问在模型训练过程中，每一次保存的pth文件是最优的吗，如果不是，如何修改呢？

ChrisChen1023 commented 4 months ago

您好，我们非常认可您的工作。请问在模型训练过程中，每一次保存的pth文件是最优的吗，如果不是，如何修改呢？

Hi,

You could validate the metrics every X iteration, and save the .pth if the value (for example PSNR) is higher. In our case, we just save the .pth for every fixed iteration, as the model collapse did not show up in our experiments. Hope this is helpful.

CoisiniStar commented 4 months ago

您好，我们非常认可您的工作。请问在模型训练过程中，每次保存的pth文件是最优的，如果不是，如何修改呢？

你好，

您可以每 X 次迭代验证一次指标，如果值（例如 PSNR）较高，则保存 .pth。在我们的例子中，我们只保存每次固定迭代的 .pth，因为模型崩溃并未在我们的实验中出现。希望这对您有所帮助。

谢谢您