search

ChrisChen1023 / HINT

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

Difference in file size between the downloaded trained model and the model I trained #13

Closed pikmin5656 closed 5 months ago

pikmin5656 commented 5 months ago

Hi. Your pretrained model downloaded from Google Drive and the model I created using the exact same code had different file sizes. Furthermore, differences in performance were also confirmed. However, there was no difference in the number of parameters, and learning was completed without any errors. What do you think this means?

Thanks.

ChrisChen1023 commented 5 months ago

Hi. Your pretrained model downloaded from Google Drive and the model I created using the exact same code had different file sizes. Furthermore, differences in performance were also confirmed. However, there was no difference in the number of parameters, and learning was completed without any errors. What do you think this means?

Thanks.

Hi,

We remove some unnecessary code before we release it, this could be the reason. As for the performance, we train the Places2 on a single A100 with around two weeks and CelebA-HQ for around one week. Performance should be the same if we use the same setting.