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

Fixs + Benchmark

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-21 15:29:34 -05:00
parent 49472adfab
commit f450fa8201
5 changed files with 60 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

9
higher/smart_aug/benchmark.py
View file

@ -8,7 +8,8 @@ from transformations import TF_loader
import torchvision.models as models
model_list={models.resnet: ['resnet18', 'resnet50','wide_resnet50_2']} #lr=0.1
#model_list={models.resnet: ['resnet18', 'resnet50','wide_resnet50_2']} #lr=0.1
model_list={models.resnet: ['resnet18']}
optim_param={
'Meta':{
@ -28,7 +29,7 @@ optim_param={
res_folder="../res/benchmark/CIFAR10/"
#res_folder="../res/HPsearch/"
epochs= 300
epochs= 400
dataug_epoch_start=0
nb_run= 1
@ -54,8 +55,8 @@ if __name__ == "__main__":
### Benchmark ###
#'''
n_inner_iter = 1
dist_mix = [0.5, 1.0]
n_inner_iter = 1#[0, 1]
dist_mix = [0.5]
N_seq_TF= [3, 4]
mag_setup = [(False, False)] #[(True, True), (False, False)] #(FxSh, Independant)