blumenstiel commited on
Commit
40a835d
·
1 Parent(s): 921b18d

Fix date from file

Browse files
Files changed (1) hide show
  1. 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="Prithvi_EO_V2_300_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_300_TL.pt",
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(