Add model classes IntelliO3_ts_architecture_finetune_all_dense,...

Add model classes IntelliO3_ts_architecture_finetune_all_dense, IntelliO3_ts_architecture_finetune_outputs and IntelliO3_ts_architecture_finetune_main_output

mlair/model_modules/model_class.py

@@ -465,7 +465,7 @@ class IntelliO3_ts_architecture(AbstractModelClass):
                                 "loss_weights": [.01, .99]
                                 }
 
-class IntelliO3_ts_architecture_freeze(IntelliO3_ts_architecture):
+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)
 
@@ -478,10 +478,31 @@ class IntelliO3_ts_architecture_freeze(IntelliO3_ts_architecture):
         for layer in self.model.layers:
             if not isinstance(layer, keras.layers.core.Dense):
                 layer.trainable = False
-    '''
+
+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.lr_decay = None
 
     def freeze_layers(self):
         for layer in self.model.layers:
             if layer.name not in ["minor_1_out_Dense", "Main_out_Dense"]:
                 layer.trainable = False
-    '''
+
\ No newline at end of file
+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.lr_decay = None
+
+    def freeze_layers(self):
+        for layer in self.model.layers:
+            if layer.name not in ["Main_out_Dense"]:
+                layer.trainable = False
\ No newline at end of file