search

liuqk3 / PUT

Paper 'Transformer based Pluralistic Image Completion with Reduced Information Loss' in TPAMI 2024 and 'Reduce Information Loss in Transformers for Pluralistic Image Inpainting' in CVPR2022
MIT License
177 stars 15 forks source link

关于训练过程中，验证集的结果 #47

Open Jackeygiaove opened 3 hours ago

Jackeygiaove commented 3 hours ago

作者您好，我用自己的训练集和验证集在训练的过程当中发现，随着epoch的进行，OUTPUT文件里的train文件下的input是不一样的，而val文件夹里的input每个epoch输出的都是相同的。但我的验证集的每一张图片确实被读入了。为什么会这样？是因为val文件夹显示input时，固定地去显示同一个batch的图片上的验证的效果吗？

liuqk3 commented 2 hours ago

是这样。因为是固定iteration（或者epoch结束后）去validation，此时会保存当前训练batch的图像，由于shuffle等操作，此次训练batch的图像和其他时候保存的不一样。对于validation数据集，没有shuffle，每次都是保存第一个batch，所以保存的图像都是一样的。具体可以看sovler的代码实现。

Jackeygiaove commented 2 hours ago

好的。谢谢您的解答！