From e34b440c6ee3ec5f43bc483707810f208ccc137e Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Thu, 18 Aug 2022 09:52:25 +0200 Subject: [PATCH] expose AutoAugmentPolicy from torchvision.prototype.transforms --- torchvision/prototype/transforms/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/prototype/transforms/__init__.py b/torchvision/prototype/transforms/__init__.py index dc6476ab4b5..ca89fee918a 100644 --- 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,