diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-09-20 11:47:24 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-09-20 11:47:24 +0200 |
| commit | 1b3b8073a19f939d18a0bb85247eb0d99284f7cc (patch) | |
| tree | e74e78230ebb179237c063fecf0b52458ce3aa3e /src/training/run_experiment.py | |
| parent | 6137f43c910946301279825e50759a9dd76c6131 (diff) | |
Bash scripts and some bug fixes.
Diffstat (limited to 'src/training/run_experiment.py')
| -rw-r--r-- | src/training/run_experiment.py | 3 |
1 files changed, 2 insertions, 1 deletions
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"]: |