[fbsync] expose AutoAugmentPolicy from torchvision.prototype.transforms (#6441)

NicolasHug · facebook-github-bot · commit 3c365cd0cf66 · 2022-08-24T04:05:22.000-07:00
Reviewed By: datumbox

Differential Revision: D38824219

fbshipit-source-id: 442e30884fe172c672efcc39efb6818f5101bda9
diff --git a/torchvision/prototype/transforms/__init__.py b/torchvision/prototype/transforms/__init__.py
@@ -3,7 +3,7 @@
 from ._transform import Transform  # usort: skip
 
 from ._augment import RandomCutmix, RandomErasing, RandomMixup
-from ._auto_augment import AugMix, AutoAugment, RandAugment, TrivialAugmentWide
+from ._auto_augment import AugMix, AutoAugment, AutoAugmentPolicy, RandAugment, TrivialAugmentWide
 from ._color import (
     ColorJitter,
     RandomAdjustSharpness,
