Fix minor bug

MTG · Mar 5, 2021 · e5b015a · e5b015a
1 parent 5786ab6
commit e5b015a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/dcase_models/model/models.py b/dcase_models/model/models.py
@@ -132,14 +132,12 @@ def build(self):
         if self.use_time_distributed:
             input_shape = (self.n_frames, self.n_freqs)
         else:
-            input_shape = (self.n_freqs)
+            input_shape = (self.n_freqs,)
 
         inputs = Input(shape=input_shape, dtype='float32', name='input')
-
+        y = inputs
         # Hidden layers
         for idx in range(len(self.hidden_layers_size)):
-            if idx == 0:
-                y = inputs
             dense_layer = Dense(self.hidden_layers_size[idx],
                                 activation=self.hidden_activation,
                                 kernel_regularizer=l2(self.l2_reg),