Merge pull request #93 from Heklis/master

Update base_model.py
BrikerMan · May 21, 2019 · c6fb74c · c6fb74c
2 parents 4dfab02 + 5ba1984
commit c6fb74c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/kashgari/tasks/classification/base_model.py b/kashgari/tasks/classification/base_model.py
@@ -184,6 +184,7 @@ def get_data_generator(self,
                     target_x.append(x[start_index: end_index])
                 target_y = y_data[start_index: end_index]
                 if len(target_x[0]) == 0:
+                    target_x.pop()
                     for x in x_data:
                         target_x.append(x[0: batch_size])
                     target_y = y_data[0: batch_size]