Update networks.py

monaanvari · Aug 9, 2022 · f911ad4 · f911ad4
1 parent 9ca099b
commit f911ad4
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/deepconsensus/models/networks.py b/deepconsensus/models/networks.py
@@ -239,10 +239,6 @@ def get_logits(self, inputs: tf.Tensor, training: bool) -> tf.Tensor:
     attention_bias = tf.expand_dims(tf.expand_dims(all_zeros, 1), 1)
 
     # Run inputs through the encoder. Encoder returns logits from dense layer.
-#     print("HERE7")
-#     inputs = tf.cast(inputs, tf.float32)
-#     print("attention_bias that goes into self.encode")
-#     print(attention_bias.dtype)
     encoder_outputs = self.encode(inputs, attention_bias, training)
     print("HERE5")
     return encoder_outputs