From 757876d57637e0da35f3680ec6ac9573b91f902a Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Mon, 3 Jul 2023 10:49:06 +0200 Subject: [PATCH] Update. --- main.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 2ed6b6b..4770a12 100755 --- a/main.py +++ b/main.py @@ -123,6 +123,8 @@ args = parser.parse_args() assert args.picocvlr_prune_properties in {"none", "train+eval", "eval"} +if args.result_dir is None: args.result_dir=f"results_{args.task}" + try: os.mkdir(args.result_dir) except FileExistsError: @@ -144,35 +146,30 @@ if args.seed >= 0: default_args = { "picoclvr": { - "result_dir": "results_picoclvr", "nb_epochs": 25, "batch_size": 25, "nb_train_samples": 250000, "nb_test_samples": 10000, }, "mnist": { - "result_dir": "results_mnist", "nb_epochs": 25, "batch_size": 10, "nb_train_samples": 250000, "nb_test_samples": 10000, }, "maze": { - "result_dir": "results_maze", "nb_epochs": 25, "batch_size": 25, "nb_train_samples": 250000, "nb_test_samples": 10000, }, "snake": { - "result_dir": "results_snake", "nb_epochs": 5, "batch_size": 25, "nb_train_samples": 250000, "nb_test_samples": 10000, }, "stack": { - "result_dir": "results_stack", "nb_epochs": 5, "batch_size": 25, "nb_train_samples": 100000, -- 2.39.5