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

4
higher/dataug.py
View file

@ -628,8 +628,8 @@ class Data_augV5(nn.Module): #Optimisation jointe (mag, proba)
if soft :
self._params['prob'].data=F.softmax(self._params['prob'].data, dim=0) #Trop 'soft', bloque en dist uniforme si lr trop faible
else:
self._params['prob'].data = F.relu(self._params['prob'].data)
#self._params['prob'].data = self._params['prob'].clamp(min=0.0,max=1.0)
#self._params['prob'].data = F.relu(self._params['prob'].data)
self._params['prob'].data = self._params['prob'].data.clamp(min=1/(self._nb_tf*100),max=1.0)
self._params['prob'].data = self._params['prob']/sum(self._params['prob']) #Contrainte sum(p)=1
if not self._fixed_mag: