From 4f94dddaaed602c384e40d0420ca6f26e938d757 Mon Sep 17 00:00:00 2001
From: "v.gramlich1" <v.gramlichfz-juelich.de>
Date: Fri, 3 Sep 2021 13:39:27 +0200
Subject: [PATCH] Set correct order in freeze layers in model_class.py
---
mlair/model_modules/model_class.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 29fe727e..f7cae7ac 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):
--
GitLab