Update model.py

model.py

@@ -192,8 +192,10 @@ class Model():
             # 最大批量大小 = 可用 GPU 内存字节数 / 4 / （张量大小 + 可训练参数）
             # max_batch_size = 10
             # Ensure batch size is within model limits:
+            print("length of inputs: ",len(inputs))
             batch_size = min(len(inputs), int(max_batch_size))
             batches = [inputs[i:i + batch_size] for i in range(0, len(inputs), batch_size)]
+            print("length of batches size: ", len(batches))
             temp_outputs = []
             processed_num = 0
             for index, batch in enumerate(batches):