Two different checkpoints for each ViT type

[hussein-jafarinia]

Two different checkpoints for each ViT type

[kevin-Abbring]

I'm curious, too. Going to test it.

[Valdiolus]

Found this

• Here we use --norm_pix_loss as the target for better representation learning. To train a baseline model (e.g., for visualization), use pixel-based construction and turn off --norm_pix_loss.

[TheodoreBagwe11]

I can only get checkpoints including encoder(mae_pretrain_vit_xx.pth), HOW to get a checkpoint including encoder and decoder (mae_visualize_vit_xx.pth) in my own model ?

[hussein-jafarinia]

I can only get checkpoints including encoder(mae_pretrain_vit_xx.pth), HOW to get a checkpoint including encoder and decoder (mae_visualize_vit_xx.pth) in my own model ?

You can find the answer in other issues. As much as I remember you should put "_full" at the end of name before ".pth".

[hussein-jafarinia]

This problem is answered in other issues completely. so I close it.