In your great work about TNT, I see transfer learning on size 384384 image based on 224224 TNT-B model. But I can't find out the method to modify the tnt.py to train TNT-B on 384384 size image with a 224224 checkpoint loaded.
I have tried to modify the conv2d in PixelEmbeded() and change img_size, but both will cause a wrong parameter number.
In your great work about TNT, I see transfer learning on size 384384 image based on 224224 TNT-B model. But I can't find out the method to modify the tnt.py to train TNT-B on 384384 size image with a 224224 checkpoint loaded.
I have tried to modify the conv2d in PixelEmbeded() and change img_size, but both will cause a wrong parameter number.
Thanks for your attention!