Call ViTMAE Forward Embedding
Hi!I would like to save the mask produced by the ViTMAEForPreTraining forward pass on one image, and apply that mask on another image. To do so, I believe that need to gain access to theforward_encoderand decoder. However, the pretrained model doesn’t have forward_encoder as an attribute. Any ideas on how to work around this?
2023-03-30T18:28:28Z
Solved this. Was looking at the FB MAE code rather than huggingface’s.