Update reader.py

paddlepalm/distribute/reader.py

@@ -125,6 +125,7 @@ def decode_fake(nums, mask, bs):
 
     n_f = len(mask) - n_t
     p1 = nums - (n_t-1) * bs
+    assert p1 % (n_f+1) == 0
     each_f = p1 // (n_f+1)
     return each_f * n_f