From 8722dbf1b2997872c698dbd2e9e47aef38464179 Mon Sep 17 00:00:00 2001
From: Xiaoyao Xi <24541791+xixiaoyao@users.noreply.github.com>
Date: Tue, 14 Apr 2020 11:47:44 +0800
Subject: [PATCH] Update reader.py

---
 paddlepalm/distribute/reader.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddlepalm/distribute/reader.py b/paddlepalm/distribute/reader.py
index 062c7c9..c9cd87a 100644
--- a/paddlepalm/distribute/reader.py
+++ b/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
 
-- 
GitLab