'Smart_aug' is module to make data_augmentation differentiable, thus allowing to learn its parameters, with 'Data_aug' classes.
Look to 'test_dataug.py'