I am trying to train resnet50 model for image classification problem.I have loaded the \'imagenet\' pretrained weights before training the model on the image dataset I have.
As an argument when creating ImageDataGenerator:
ImageDataGenerator
train_datagen = ImageDataGenerator(preprocessing_function=preprocess_input)