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

Borne mag + Regularisation mag

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-19 15:37:29 -05:00
parent f4bdd9bca5
commit 64282bda3a
10 changed files with 43 additions and 228 deletions
Download patch file Download diff file Expand all files Collapse all files

5
higher/utils.py
View file

@ -254,6 +254,11 @@ def print_torch_mem(add_info=''):
torch.cuda.max_memory_cached()/ mega_bytes)
print(string)
def TF_influence(log):
proba=[[x["param"][idx]['p'] for x in log] for idx, _ in enumerate(log[0]["param"])]
mag=[[x["param"][idx]['m'] for x in log] for idx, _ in enumerate(log[0]["param"])]
return np.mean(proba, axis=1)*np.mean(mag, axis=1) #Pourrait etre interessant de multiplier avant le mean
class loss_monitor(): #Voir https://github.com/pytorch/ignite
def __init__(self, patience, end_train=1):