Update action_predict.py

javierlorenzod · Jul 5, 2023 · 1be22c4 · 1be22c4
1 parent 6d598a4
commit 1be22c4
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/action_predict.py b/action_predict.py
@@ -3318,12 +3318,12 @@ def get_model(self, data_params):
             tf_enc_inputs = layers.Concatenate(axis=-1)(network_inputs[1:])
         else:
             tf_enc_inputs = network_inputs[1]
-            tf_features = Encoder(num_layers=num_layers, d_model=d_model, num_heads=num_heads,
-                                    dff=dff, rate=dropout_rate)(tf_enc_inputs,None)
-            #tf_features = Dense(name='final_emb_' + data_types[i],units=attention_size)(tf_features)
-            encoder_outputs.append(
-                tf_features
-                    )
+        tf_features = Encoder(num_layers=num_layers, d_model=d_model, num_heads=num_heads,
+                                dff=dff, rate=dropout_rate)(tf_enc_inputs,None)
+        #tf_features = Dense(name='final_emb_' + data_types[i],units=attention_size)(tf_features)
+        encoder_outputs.append(
+            tf_features
+                )
 
         if len(encoder_outputs) > 1:
             att_enc_out = []