Ошибка в bottleneck_model.predict_generator

Question

Я пытаюсь запустить код чтения по губам из https://github.com/zrobert7/cs231N_lip_reading/blob/master/main.py

Здесь автор создал файл bottleneck_feature.npy, используя

bottleneck_features_train = bottleneck_model.predict_generator(self.training_generator(),steps=np.shape(self.X_train)[0] / self.config.batch_size)

, где training_generator () равен

def training_generator(self):
while True:
for i in range(int(np.shape(self.X_train)[0]/ self.config.batch_size)):
  x = self.X_train[i * self.config.batch_size: (i + 1) * self.config.batch_size]
  y = self.y_train[i * self.config.batch_size: (i + 1) * self.config.batch_size]
  one_hot_labels_train = keras.utils.to_categorical(y, num_classes=self.config.num_classes)
  yield (x, one_hot_labels_train)

Останавливается внезапно без ошибки.Пожалуйста, помогите