Skip to content
Snippets Groups Projects
Commit 4f94ddda authored by v.gramlich1's avatar v.gramlich1
Browse files

Set correct order in freeze layers in model_class.py

parent 6d39cf45
No related branches found
No related tags found
1 merge request!302Draft: Resolve "Class-based Oversampling technique"
Pipeline #81559 passed
Stage: test
Stage: docs
Stage: pages
Stage: deploy
Show whitespace changes
Inline Side-by-side
mlair/model_modules/model_class.py
+ 3
− 3
View file @ 4f94ddda
......@@ -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):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment