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

Minimum de proba !=0 (ReLu=>Clamp)

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-01-13 13:26:53 -05:00
parent 18fe14079a
commit 9693dd7113
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
higher/train_utils.py
View file

@ -735,6 +735,8 @@ def run_dist_dataugV2(model, opt_param, epochs=1, inner_it=0, dataug_epoch_start
val_loss.backward()
#print("meta", time.process_time()-t)
#print('proba grad',model['data_aug']['prob'].grad)
if model['data_aug']['prob'].grad is None or model['data_aug']['mag'] is None:
print("Warning no grad (iter",i,") :\n Prob-",model['data_aug']['prob'].grad,"\n Mag-", model['data_aug']['mag'].grad)
countcopy+=1
model_copy(src=fmodel, dst=model)