Я пытаюсь запустить код чтения по губам из 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)
Останавливается внезапно без ошибки.Пожалуйста, помогите