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

Ajout meta-learning differee

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-19 11:54:54 -05:00
parent f2cf244801
commit 53f6600ff6
4 changed files with 70 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

6
higher/smart_aug/benchmark.py
View file

@ -4,6 +4,7 @@
from dataug import *
#from utils import *
from train_utils import *
from transformations import TF_loader
import torchvision.models as models
@ -13,6 +14,7 @@ optim_param={
'Meta':{
'optim':'Adam',
'lr':1e-2, #1e-2
'epoch_start': 2, #0 / 2 (Resnet?)
},
'Inner':{
'optim': 'SGD',
@ -26,9 +28,9 @@ optim_param={
res_folder="../res/benchmark/CIFAR10/"
#res_folder="../res/HPsearch/"
epochs= 200
epochs= 300
dataug_epoch_start=0
nb_run= 3
nb_run= 1
tf_config='../config/base_tf_config.json'
TF_loader=TF_loader()