I trained a convolutional neural network with a training_batch_size of 17000 images and a validation_batch_size of 3400 images. My dataset (images) are divided in 6 classes.