diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 29fe727e969a23f2a625779e0669880bea1fc52e..f7cae7ac640e2ab412096ff1859ed10b4c7c761b 100644
--- a/mlair/model_modules/model_class.py
+++ b/mlair/model_modules/model_class.py
@@ -469,9 +469,9 @@ class IntelliO3_ts_architecture_finetune_all_dense(IntelliO3_ts_architecture):
     def __init__(self, input_shape: list, output_shape: list):
         super().__init__(input_shape, output_shape)
 
-        self.freeze_layers()
         self.initial_lr = 1e-5
         self.apply_to_model()
+        self.freeze_layers()
         # self.lr_decay = None
 
     def freeze_layers(self):
@@ -483,9 +483,9 @@ class IntelliO3_ts_architecture_finetune_outputs(IntelliO3_ts_architecture):
     def __init__(self, input_shape: list, output_shape: list):
         super().__init__(input_shape, output_shape)
 
-        self.freeze_layers()
         self.initial_lr = 1e-5
         self.apply_to_model()
+        self.freeze_layers()
         # self.lr_decay = None
 
     def freeze_layers(self):
@@ -497,9 +497,9 @@ class IntelliO3_ts_architecture_finetune_main_output(IntelliO3_ts_architecture):
     def __init__(self, input_shape: list, output_shape: list):
         super().__init__(input_shape, output_shape)
 
-        self.freeze_layers()
         self.initial_lr = 1e-5
         self.apply_to_model()
+        self.freeze_layers()
         # self.lr_decay = None
 
     def freeze_layers(self):