Fix location of ccs bq embedding layer.

PiperOrigin-RevId: 506648389
google · Feb 2, 2023 · 02274ba · 02274ba
1 parent 63df275
commit 02274ba
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/deepconsensus/models/networks.py b/deepconsensus/models/networks.py
@@ -439,13 +439,6 @@ def encode(self, inputs: tf.Tensor, attention_bias: tf.Tensor,
         embedded = self.ip_embedding_layer(tf.cast(inputs[:, :, i], tf.int32))
         embedded_inputs.append(embedded)
 
-    if self.params.use_ccs_bq:
-      for i in range(*ccs_bq_indices):
-        # Add 1 to ccs base quality scores to shift gaps from -1 to 0.
-        embedded = self.ccs_base_quality_scores_embedding_layer(
-            tf.cast(inputs[:, :, i] + 1, tf.int32))
-        embedded_inputs.append(embedded)
-
     if self.params.use_strand:
       for i in range(*strand_indices):
         embedded = self.strand_embedding_layer(
@@ -458,6 +451,13 @@ def encode(self, inputs: tf.Tensor, attention_bias: tf.Tensor,
             tf.cast(inputs[:, :, i], tf.int32))
         embedded_inputs.append(embedded)
 
+    if self.params.use_ccs_bq:
+      for i in range(*ccs_bq_indices):
+        # Add 1 to ccs base quality scores to shift gaps from -1 to 0.
+        embedded = self.ccs_base_quality_scores_embedding_layer(
+            tf.cast(inputs[:, :, i] + 1, tf.int32))
+        embedded_inputs.append(embedded)
+
     if self.params.use_sn:
       # The last four elements in the last dimension in the inputs tensor
       # correspond to the four signal-to-noise ratio scores for A, G, C, T.