fix assertion for even number of batch size

zihangdai · Jul 4, 2019 · 7ef2f2d · 7ef2f2d · vochicong · Oct 8, 2019
1 parent 4c574d6
commit 7ef2f2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modeling.py b/modeling.py
@@ -233,7 +233,7 @@ def relative_positional_encoding(qlen, klen, d_model, clamp_len, attn_type,
 
     if bsz is not None:
       # With bi_data, the batch size should be divisible by 2.
-      assert bsz%2 == 0
+      tf.debugging.assert_equal(bsz % 2, 0)
       fwd_pos_emb = positional_embedding(fwd_pos_seq, inv_freq, bsz//2)
       bwd_pos_emb = positional_embedding(bwd_pos_seq, inv_freq, bsz//2)
     else: