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-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

5
higher/smart_aug/test_dataug.py
View file

@ -6,10 +6,10 @@ from LeNet import *
from dataug import *
#from utils import *
from train_utils import *
#from transformations import TF_loader
from transformations import TF_loader
postfix=''
TF_loader=TF.TF_loader()
TF_loader=TF_loader()
device = torch.device('cuda') #Select device to use
@ -40,6 +40,7 @@ if __name__ == "__main__":
'Meta':{
'optim':'Adam',
'lr':1e-2, #1e-2
'epoch_start': 2, #0 / 2 (Resnet?)
},
'Inner':{
'optim': 'SGD',