diff --git a/vissl/data/ssl_transforms/rand_auto_aug.py b/vissl/data/ssl_transforms/rand_auto_aug.py index 88a691a56..b3fd78c04 100644 --- a/vissl/data/ssl_transforms/rand_auto_aug.py +++ b/vissl/data/ssl_transforms/rand_auto_aug.py @@ -72,7 +72,7 @@ # Modification/Addition -@register_transform("RandAugment") +@register_transform("RandAugment", bypass_checks=True) class RandAugment(ClassyTransform): """ Create a RandAugment transform.