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

Debut implementation Dataugv5

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-14 21:42:00 -05:00
parent 103277fadd
commit 05f81787d6
3 changed files with 31 additions and 58 deletions
Download patch file Download diff file Expand all files Collapse all files

1
higher/train_utils.py
View file

@ -649,6 +649,7 @@ def run_dist_dataugV2(model, epochs=1, inner_it=0, dataug_epoch_start=0, print_f
print('Data Augmention : {} (Epoch {})'.format(model._data_augmentation, dataug_epoch_start))
print('TF Proba :', model['data_aug']['prob'].data)
#print('proba grad',aug_model['data_aug']['prob'].grad)
print('TF Mag :', model['data_aug']['mag'].data)
#############
#### Log ####
data={