From 7330792e4ad6997225778f032906cd8e8b2c25b1 Mon Sep 17 00:00:00 2001
From: Charly Lamothe <charly.lamothe@univ-amu.fr>
Date: Thu, 19 Dec 2019 12:58:29 +0100
Subject: [PATCH] Set dataset_name arg in compute_results to required (tmp)
---
code/compute_results.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/code/compute_results.py b/code/compute_results.py
index e9c93e1..40e1371 100644
--- a/code/compute_results.py
+++ b/code/compute_results.py
@@ -121,7 +121,7 @@ if __name__ == "__main__":
parser.add_argument('--stage', nargs='?', type=int, required=True, help='Specify the stage number among [1, 4].')
parser.add_argument('--experiment_ids', nargs='+', type=int, required=True, help='Compute the results of the specified experiment id(s).' + \
'stage=1: {{base_with_params}} {{random_with_params}} {{omp_with_params}} {{base_wo_params}} {{random_wo_params}} {{omp_wo_params}}')
- parser.add_argument('--dataset_name', nargs='?', type=str, help='Specify the dataset name. TODO: read it from models dir directly.')
+ parser.add_argument('--dataset_name', nargs='?', type=str, required=True, help='Specify the dataset name. TODO: read it from models dir directly.')
parser.add_argument('--results_dir', nargs='?', type=str, default=DEFAULT_RESULTS_DIR, help='The output directory of the results.')
parser.add_argument('--models_dir', nargs='?', type=str, default=DEFAULT_MODELS_DIR, help='The output directory of the trained models.')
args = parser.parse_args()
--
GitLab