diff --git a/GANDLF/models/__init__.py b/GANDLF/models/__init__.py index 714baca12..4faa1960b 100644 --- a/GANDLF/models/__init__.py +++ b/GANDLF/models/__init__.py @@ -18,7 +18,7 @@ efficientnetB6, efficientnetB7, ) -from .imagenet import ( +from .imagenet_vgg import ( imagenet_vgg11, imagenet_vgg11_bn, imagenet_vgg13, diff --git a/GANDLF/models/imagenet.py b/GANDLF/models/imagenet_vgg.py similarity index 100% rename from GANDLF/models/imagenet.py rename to GANDLF/models/imagenet_vgg.py