No module named 'efficientnet.layers'

[Samjith888]

Got this error while loading a trained model

from efficientnet.layers import Swish, DropConnect ModuleNotFoundError: No module named 'efficientnet.layers'

script used :

from efficientnet.layers import Swish, DropConnect
from efficientnet.model import ConvKernalInitializer
from tensorflow.keras.utils import get_custom_objects

get_custom_objects().update({
    'ConvKernalInitializer': ConvKernalInitializer,
    'Swish': Swish,
    'DropConnect':DropConnect
})
from tensorflow.keras.models import load_model
model = load_model("./models/cats_and_dogs_small.h5")

Installed efficientnet using following command

pip install -U efficientnet