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

inference lack module #34

Open Damehou opened 3 months ago

Damehou commented 3 months ago
截屏2024-05-16 15 16 33
feifeiwei commented 3 months ago

+1

liuqk3 commented 3 months ago

Thanks for pointing out this bug. I have fixed this. Please update your code with this repo.

However, You can also comment this line for the reason that the PartialConv2d is not used by default.