eliasbela commited on
Commit
7dc0c59
1 Parent(s): 7b231fe

Change import of MAE VIT

Browse files

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.

Files changed (1) hide show
  1. Prithvi_run_inference.py +1 -2
Prithvi_run_inference.py CHANGED
@@ -9,8 +9,7 @@ import torch
9
  import yaml
10
  from einops import rearrange
11
 
12
- from mae.models_mae import MaskedAutoencoderViT
13
-
14
 
15
  NO_DATA = -9999
16
  NO_DATA_FLOAT = 0.0001
 
9
  import yaml
10
  from einops import rearrange
11
 
12
+ from Prithvi import MaskedAutoencoderViT
 
13
 
14
  NO_DATA = -9999
15
  NO_DATA_FLOAT = 0.0001