Patches done on FBCode sync (#3204)

* Patches required for FBCode merge. * Patching quantization model imports. * import QuantizableMobileNetV2 * Adding newline to avoid lit errors

Patches done on FBCode sync (#3204)
* Patches required for FBCode merge. * Patching quantization model imports. * import QuantizableMobileNetV2 * Adding newline to avoid lit errors
8c0bb6c2 · Vasilis Vryniotis · GitHub · 3d60f498 · 8c0bb6c2 · 8c0bb6c2
Unverified Commit 8c0bb6c2 authored Dec 23, 2020 by Vasilis Vryniotis Committed by GitHub Dec 23, 2020
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

torchvision/models/mobilenet.py torchvision/models/mobilenet.py +3 -1

torchvision/models/quantization/mobilenet.py torchvision/models/quantization/mobilenet.py +3 -1

No files found.
--- a/torchvision/models/mobilenet.py
+++ b/torchvision/models/mobilenet.py
-from .mobilenetv2 import MobileNetV2, mobilenet_v2
+from .mobilenetv2 import MobileNetV2, mobilenet_v2, __all__ as mv2_all
+
+__all__ = mv2_all
--- a/torchvision/models/quantization/mobilenet.py
+++ b/torchvision/models/quantization/mobilenet.py
-from .mobilenetv2 import MobileNetV2, mobilenet_v2
+from .mobilenetv2 import QuantizableMobileNetV2, mobilenet_v2, __all__ as mv2_all
+
+__all__ = mv2_all