diff options
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. |