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
173 stars 15 forks source link

训练问题 #32

Open zhangshilong2000 opened 7 months ago

zhangshilong2000 commented 7 months ago

您好，我看到训练的时候，也会进行验证，能否使用自己的掩码进行验证？

liuqk3 commented 7 months ago

可以的。需要按照说明准备好相应的数据，在config文件中的dataloader.validation_datasets中进行配置相应的dataset即可。