From 48d46ee8993bd117162b12bc4ecb327acc28fad9 Mon Sep 17 00:00:00 2001
From: Vasilis Vryniotis <vvryniotis@fb.com>
Date: Wed, 8 Sep 2021 10:42:41 -0700
Subject: [PATCH] Adding RandAugment implementation (#4348) (#421)

Summary:
Pull Request resolved: https://github.com/facebookresearch/vissl/pull/421

* Adding randaugment implementation

* Refactoring.

* Adding num_magnitude_bins.

* Adding FIXME.

Reviewed By: fmassa

Differential Revision: D30793331

fbshipit-source-id: 384d9b28dd5a695654bd3f6b0fe8ae4f587d6932
---
 vissl/data/ssl_transforms/rand_auto_aug.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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.