ModuleNotFoundError: No module named 'keras.applications.inceptionresnetv2'

[zotttttttt]

Hey! Great code by the way, but I want to point out an error. If I try to use InceptionResnetV2, I get "ModuleNotFoundError: No module named 'keras.applications.inceptionresnetv2'"

To fix that change

keras.applications.inceptionresnetv2

to

keras.application.inception_resnet_v2

in your main.py so that it would look like that:

elif args.model == "InceptionResNetV2":
    from keras.applications.inception_resnet_v2 import preprocess_input
    preprocessing_function = preprocess_input
    base_model = InceptionResNetV2(weights='imagenet', include_top=False, input_shape=(HEIGHT, WIDTH, 3))