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.

IBM NASA Geospatial org

Thanks for spotting this!

Paolo-Fraccaro changed pull request status to merged

Sign up or log in to comment