blumenstiel commited on
Commit
ff853ee
·
1 Parent(s): b0da51e

Update inference

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -494,7 +494,7 @@ if __name__ == "__main__":
494
  parser.add_argument(
495
  "--checkpoint",
496
  type=str,
497
- default="Prithvi_EO_V2_tiny.pt",
498
  help="Path to a checkpoint file to load from.",
499
  )
500
  parser.add_argument(
 
494
  parser.add_argument(
495
  "--checkpoint",
496
  type=str,
497
+ default="Prithvi_EO_V2_tiny_TL.pt",
498
  help="Path to a checkpoint file to load from.",
499
  )
500
  parser.add_argument(