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

Transfert TF loader vers transformations.py

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-19 09:30:16 -05:00
parent d53b385c43
commit f2cf244801
3 changed files with 123 additions and 123 deletions
Download patch file Download diff file Expand all files Collapse all files

3
higher/smart_aug/test_dataug.py
View file

@ -6,9 +6,10 @@ from LeNet import *
from dataug import *
#from utils import *
from train_utils import *
#from transformations import TF_loader
postfix=''
TF_loader=TF_loader()
TF_loader=TF.TF_loader()
device = torch.device('cuda') #Select device to use