From 25b5d6983d51e0e791b96a76beb7e49f392cd9a8 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Mon, 7 Dec 2020 22:54:04 +0100
Subject: Segmentation working!

---
 src/training/run_experiment.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'src/training/run_experiment.py')

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.
-- 
cgit v1.2.3-70-g09d2