From 368baf25c8c0f8b5b44379623ef059540986ef13 Mon Sep 17 00:00:00 2001 From: Charles Gaydon Date: Thu, 12 Oct 2023 17:51:20 +0200 Subject: [PATCH] Correct default model name --- run.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/run.py b/run.py index 341a99ee..5f7f03bf 100755 --- 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"