Change import of MAE VIT
#10
by
eliasbela
- opened
In Prithvi_run_inference.py the MaskedAutoencoderViT gets imported from mae.models_mae (the original mae repo I assume) while it seems like it should get imported from Prithvi.py since it got altered w.r.t the implementation of the original repo.
Thanks for spotting this!
Paolo-Fraccaro
changed pull request status to
merged