Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "fdf55b1f1c25c7e2fedcdd0dc9c6f3db005453bb"
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 ...@@ -3,7 +3,7 @@ from . import functional # usort: skip
from ._transform import Transform # usort: skip from ._transform import Transform # usort: skip
from ._augment import RandomCutmix, RandomErasing, RandomMixup 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 ( from ._color import (
ColorJitter, ColorJitter,
RandomAdjustSharpness, 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