Correct default model name

IGNF · Oct 12, 2023 · 368baf2 · 368baf2
1 parent 4817948
commit 368baf2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/run.py b/run.py
@@ -22,8 +22,10 @@
 
 TASK_NAME_DETECTION_STRING = "task.task_name="
 DEFAULT_DIRECTORY = "trained_model_assets/"
-DEFAULT_CONFIG_FILE = "proto151_V2.0_epoch_100_Myria3DV3.1.0_predict_config_V3.5.0.yaml"
-DEFAULT_CHECKPOINT = "proto151_V2.0_epoch_100_Myria3DV3.1.0.ckpt"
+DEFAULT_CONFIG_FILE = (
+    "20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml"
+)
+DEFAULT_CHECKPOINT = "20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0.ckpt"
 DEFAULT_ENV = "placeholder.env"