checking mini-bug parameters to lambda layer

Machine-Listeners-Valencia · Dec 4, 2020 · 8e11dc6 · 8e11dc6
1 parent 5355205
commit 8e11dc6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/code/models.py b/code/models.py
@@ -51,7 +51,8 @@ def res_conv_standard_post_csse_split_freqs(h, w, n_channels, n_classes,
 
     if n_split_freqs == 2:
 
-        splits = keras.layers.Lambda(freq_split)(ip, n_split_freqs, f_split_freqs)
+        splits = keras.layers.Lambda(freq_split, arguments={'n_split_freqs': n_split_freqs,
+                                                            'f_split_freqs': f_split_freqs})(ip)
 
         x1 = splits[0]
         x2 = splits[1]
@@ -86,12 +87,13 @@ def res_conv_standard_post_csse_split_freqs(h, w, n_channels, n_classes,
 
     elif n_split_freqs == 3:
 
-        splits = keras.layers.Lambda(freq_split, arguments={'n_split_freqs': n_split_freqs, 'f_split_freqs': f_split_freqs})(ip)
+        splits = keras.layers.Lambda(freq_split, arguments={'n_split_freqs': n_split_freqs,
+                                                            'f_split_freqs': f_split_freqs})(ip)
 
         x1 = splits[0]
         x2 = splits[1]
         x3 = splits[2]
-        
+
         for i in range(0, len(nfilters)):
             x1 = network_module(x1, nfilters[i], ratio, pools_size[i], dropouts_rate[i])
             x2 = network_module(x2, nfilters[i], ratio, pools_size[i], dropouts_rate[i])