diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-12-07 22:54:04 +0100 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-12-07 22:54:04 +0100 |
| commit | 25b5d6983d51e0e791b96a76beb7e49f392cd9a8 (patch) | |
| tree | 526ba739714b3d040f7810c1a6be3ff0ba37fdb1 /src/training/run_experiment.py | |
| parent | 5529e0fc9ca39e81fe0f08a54f257d32f0afe120 (diff) | |
Segmentation working!
Diffstat (limited to 'src/training/run_experiment.py')
| -rw-r--r-- | src/training/run_experiment.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/training/run_experiment.py b/src/training/run_experiment.py index a883b45..0167725 100644 --- a/src/training/run_experiment.py +++ b/src/training/run_experiment.py @@ -286,6 +286,7 @@ def run_experiment( callbacks=callbacks, transformer_model=experiment_config["train_args"]["transformer_model"], max_norm=experiment_config["train_args"]["max_norm"], + freeze_backbone=experiment_config["train_args"]["freeze_backbone"], ) # Train the model. |