Я хочу тренировать глубокую сеть, начиная со следующего слоя:

model = Sequential()
model.add(Conv2D(32, 3, 3, input_shape=(32, 32, 3)))

с помощью

history = model.fit_generator(get_training_data(),
                samples_per_epoch=1, nb_epoch=1,nb_val_samples=5,
                verbose=1,validation_data=get_validation_data()

со следующим генератором:

def get_training_data(self):
     while 1:
        for i in range(1,5):
            image = self.X_train[i]
            label = self.Y_train[i]
            yield (image,label)

(генератор валидации выглядит аналогично).

Во время обучения я получаю ошибку:

Error when checking model input: expected convolution2d_input_1 to have 4 
dimensions, but got array with shape (32, 32, 3)

Как это может быть, с первым слоем

 model.add(Conv2D(32, 3, 3, input_shape=(32, 32, 3)))

?