[jacinto-ai/pytorch-jacinto-ai-devkit.git] / modules / pytorch_jacinto_ai / vision / models / mobilenetv2.py
diff --git a/modules/pytorch_jacinto_ai/vision/models/mobilenetv2.py b/modules/pytorch_jacinto_ai/vision/models/mobilenetv2.py
index a47d4b135fd170801195a55cb6dc92d7590c418a..89b59e2331ce24bbf64053394891223a0b50db81 100644 (file)
from ... import xnn
###################################################
-__all__ = ['MobileNetV2TVBase', 'MobileNetV2TV', 'mobilenet_v2_tv', 'get_config']
+__all__ = ['MobileNetV2TVBase', 'MobileNetV2TV', 'mobilenet_v2', 'mobilenet_v2_tv', 'get_config']
###################################################
state_dict = load_state_dict_from_url(model_urls['mobilenet_v2_tv'], progress=progress)
model.load_state_dict(state_dict)
return model
+
+mobilenet_v2 = mobilenet_v2_tv
\ No newline at end of file