search

XPixelGroup / HAT

CVPR2023 - Activating More Pixels in Image Super-Resolution Transformer Arxiv - HAT: Hybrid Attention Transformer for Image Restoration
Apache License 2.0
1.14k stars 134 forks source link

For training HAT-GAN what pretrained model should I use? #104

Open Ar0Kim opened 8 months ago

Ar0Kim commented 8 months ago

I want to train HAT-GAT using HAT_GAN_REAL_SRx4.yml and for 'pretrain_network_g' what pretrained model should I use and where i can get it from?

please tell me thank you

chxy95 commented 8 months ago

@Ar0Kim The mse-based real-HAT model is not saved now. Use train_Real_HAT_SRx4_mse_model.yml to train the 'pretrain_network_g' that you need.

Ar0Kim commented 8 months ago

Thank you so much.