Skip to content

Commit

Permalink
Merge branch 'model-refactor' of https://github.com/rvankoert/loghi-htr
Browse files Browse the repository at this point in the history
… into model-refactor
  • Loading branch information
TimKoornstra committed Aug 17, 2023
2 parents 5d3f050 + d9d0158 commit c3c5aaf
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions src/vgsl_model_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,9 +85,9 @@ def init_model_from_string(self, vgsl_spec_string,
setattr(self, f"avgpool{index}", self.avgpool_generator(layer))
self.history.append(f"avgpool{index}")
elif layer.startswith('RB'):
setattr(self, f"resLrock{index}",
self.resLrock_generator(layer))
self.history.append(f"resLrock{index}")
setattr(self, f"ResidualBlock{index}",
self.residual_block_generator(layer))
self.history.append(f"ResidualBlock{index}")
elif layer.startswith('D'):
setattr(self, f"dropout{index}", self.dropout_generator(layer))
self.history.append(f"dropout{index}")
Expand Down Expand Up @@ -316,7 +316,7 @@ def lstm_generator(self, layer):
"kernel_initializer": self.initializer
}

rnn_layer = layers.GRU
rnn_layer = layers.LSTM
return rnn_layer(**kwargs)

def gru_generator(self, layer):
Expand Down Expand Up @@ -361,7 +361,7 @@ def bidirectional_generator(self, layer):

return layers.Bidirectional(rnn_layer(**rnn_params), merge_mode='concat')

def resLrock_generator(self, layer):
def residual_block_generator(self, layer):
"""
Create a Residual Block with Conv2D layers and an elu BatchNorm, RB
"""
Expand Down

0 comments on commit c3c5aaf

Please sign in to comment.