Skip to content

GitLab

Unverified Commit c3dc2552 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

expose AutoAugmentPolicy from torchvision.prototype.transforms (#6441)

parent 162267ca
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/prototype/transforms/__init__.py
View file @ c3dc2552
......@@ -3,7 +3,7 @@ from . import functional # usort: skip
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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment