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

Ajout Meta-scheduler a run_dist_dataugV3

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-26 12:19:44 -05:00
parent 2cbe3d09aa
commit e2691a1c38
2 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
higher/smart_aug/test_dataug.py
View file

@ -8,7 +8,7 @@ from dataug import *
from train_utils import *
from transformations import TF_loader
postfix=''
postfix='-metaScheduler'
TF_loader=TF_loader()
device = torch.device('cuda') #Select device to use
@ -40,9 +40,10 @@ if __name__ == "__main__":
optim_param={
'Meta':{
'optim':'Adam',
'lr':1e-2, #1e-2
'lr':1e-4, #1e-2
'epoch_start': 2, #0 / 2 (Resnet?)
'reg_factor': 0.001,
'scheduler': 'multiStep', #None, 'multiStep'
},
'Inner':{
'optim': 'SGD',
@ -138,7 +139,7 @@ if __name__ == "__main__":
inner_it=n_inner_iter,
dataug_epoch_start=dataug_epoch_start,
opt_param=optim_param,
print_freq=10,
print_freq=20,
unsup_loss=1,
hp_opt=False,
save_sample_freq=None)