diff --git a/torchvision/models/mobilenet.py b/torchvision/models/mobilenet.py index 8be33d71a4e..0c7cc10df5b 100644 --- a/torchvision/models/mobilenet.py +++ b/torchvision/models/mobilenet.py @@ -1 +1,3 @@ -from .mobilenetv2 import MobileNetV2, mobilenet_v2 +from .mobilenetv2 import MobileNetV2, mobilenet_v2, __all__ as mv2_all + +__all__ = mv2_all diff --git a/torchvision/models/quantization/mobilenet.py b/torchvision/models/quantization/mobilenet.py index 8be33d71a4e..8b4e61f8ae6 100644 --- a/torchvision/models/quantization/mobilenet.py +++ b/torchvision/models/quantization/mobilenet.py @@ -1 +1,3 @@ -from .mobilenetv2 import MobileNetV2, mobilenet_v2 +from .mobilenetv2 import QuantizableMobileNetV2, mobilenet_v2, __all__ as mv2_all + +__all__ = mv2_all diff --git a/version.txt b/version.txt index ddf3a6311c3..657e7c07f92 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -0.9.0a0 \ No newline at end of file +0.9.0a0