diff --git a/efficientnet/__init__.py b/efficientnet/__init__.py index 44a90ad..19cdeb3 100644 --- a/efficientnet/__init__.py +++ b/efficientnet/__init__.py @@ -59,7 +59,7 @@ def wrapper(*args, **kwargs): return wrapper -def init_keras_custom_objects(): +def init_tfkeras_custom_objects(): import keras from . import model