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

Modif solarize (Tjrs pas differentiable...)

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-27 17:19:51 -05:00
parent 4a7e73088d
commit d822f8f92e
5 changed files with 56 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

2
higher/train_utils.py
View file

@ -651,7 +651,7 @@ def run_dist_dataugV2(model, epochs=1, inner_it=0, dataug_epoch_start=0, print_f
print('TF Proba :', model['data_aug']['prob'].data)
#print('proba grad',model['data_aug']['prob'].grad)
print('TF Mag :', model['data_aug']['mag'].data)
#print('Mag grad',model['data_aug']['mag'].grad)
print('Mag grad',model['data_aug']['mag'].grad)
#print('Reg loss:', model['data_aug'].reg_loss().item())
#############
#### Log ####