Commit
·
40a835d
1
Parent(s):
921b18d
Fix date from file
Browse files- inference.py +3 -3
inference.py
CHANGED
@@ -140,7 +140,7 @@ def load_example(
|
|
140 |
location_coords.append(coords)
|
141 |
|
142 |
try:
|
143 |
-
match = re.search(r'(\d{7}T\d{6})', file)
|
144 |
if match:
|
145 |
year = int(match.group(1)[:4])
|
146 |
julian_day = match.group(1).split('T')[0][4:]
|
@@ -485,13 +485,13 @@ if __name__ == "__main__":
|
|
485 |
"--config_path",
|
486 |
"-c",
|
487 |
type=str,
|
488 |
-
default="
|
489 |
help="Path to yaml file containing model training parameters.",
|
490 |
)
|
491 |
parser.add_argument(
|
492 |
"--checkpoint",
|
493 |
type=str,
|
494 |
-
default="
|
495 |
help="Path to a checkpoint file to load from.",
|
496 |
)
|
497 |
parser.add_argument(
|
|
|
140 |
location_coords.append(coords)
|
141 |
|
142 |
try:
|
143 |
+
match = re.search(r'(\d{7,8}T\d{6})', file)
|
144 |
if match:
|
145 |
year = int(match.group(1)[:4])
|
146 |
julian_day = match.group(1).split('T')[0][4:]
|
|
|
485 |
"--config_path",
|
486 |
"-c",
|
487 |
type=str,
|
488 |
+
default="Prithvi_EO_V2_300M_TL_config.yaml",
|
489 |
help="Path to yaml file containing model training parameters.",
|
490 |
)
|
491 |
parser.add_argument(
|
492 |
"--checkpoint",
|
493 |
type=str,
|
494 |
+
default="Prithvi_EO_V2_300M_TL.pt",
|
495 |
help="Path to a checkpoint file to load from.",
|
496 |
)
|
497 |
parser.add_argument(
|