facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Other
6.93k stars 1.17k forks source link

Finetune ViT-huge with 448 size #110

Open liuxingbin opened 1 year ago

liuxingbin commented 1 year ago

Hi! When I reproduce the result of ViT-huge with 448 size. I get nan problem. What's the parameter setting for 448? Looking forward to your reply.