Training the VGG16 model