From 1b3b8073a19f939d18a0bb85247eb0d99284f7cc Mon Sep 17 00:00:00 2001 From: aktersnurra Date: Sun, 20 Sep 2020 11:47:24 +0200 Subject: Bash scripts and some bug fixes. --- src/training/run_experiment.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/training/run_experiment.py') diff --git a/src/training/run_experiment.py b/src/training/run_experiment.py index cc882ad..9d45841 100644 --- a/src/training/run_experiment.py +++ b/src/training/run_experiment.py @@ -273,7 +273,8 @@ def run_experiment( ) # Train the model. - trainer.fit(model) + if experiment_config["train"]: + trainer.fit(model) # Run inference over test set. if experiment_config["test"]: -- cgit v1.2.3-70-g09d2