Skip to content

Commit 4243f07

Browse files
committed
Adding RandAugment to AutoAugment impl, some tweaks to AA included
1 parent 4748c6d commit 4243f07

File tree

3 files changed

+255
-72
lines changed

3 files changed

+255
-72
lines changed

timm/data/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,5 @@
44
from .transforms import *
55
from .loader import create_loader, create_transform
66
from .mixup import mixup_target, FastCollateMixup
7+
from .auto_augment import RandAugment, AutoAugment, rand_augment_ops, auto_augment_policy,\
8+
rand_augment_transform, auto_augment_transform

0 commit comments

Comments
 (0)