Change import of MAE VIT
Browse filesIn 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.
- 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
|
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
|