AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

Changement mesure memoire + Tests solarize differentiable

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-10 14:36:12 -05:00
parent 6277e268c1
commit 7d5aa7c6fb
4 changed files with 102 additions and 66 deletions
Download patch file Download diff file Expand all files Collapse all files

6
higher/smart_aug/datasets.py
View file

@ -39,14 +39,14 @@ transform_train = torchvision.transforms.Compose([
])
## RandAugment ##
from RandAugment import RandAugment
#from RandAugment import RandAugment
# Add RandAugment with N, M(hyperparameter)
rand_aug={'N': 2, 'M': 1}
#rand_aug={'N': 2, 'M': 1}
#rand_aug={'N': 2, 'M': 9./30} #RN-ImageNet
#rand_aug={'N': 3, 'M': 5./30} #WRN-CIFAR10
#rand_aug={'N': 2, 'M': 14./30} #WRN-CIFAR100
#rand_aug={'N': 3, 'M': 7./30} #WRN-SVHN
transform_train.transforms.insert(0, RandAugment(n=rand_aug['N'], m=rand_aug['M']))
#transform_train.transforms.insert(0, RandAugment(n=rand_aug['N'], m=rand_aug['M']))
### Classic Dataset ###