AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-03 11:40:46 +02:00
Code Issues Projects Releases Packages Wiki Activity

Pas reg sur apprentissage mag seules

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-03-05 15:35:22 -05:00
parent 4166922c34
commit b820f49437
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
higher/smart_aug/dataug.py
View file

@ -251,7 +251,7 @@ class Data_augV5(nn.Module): #Optimisation jointe (mag, proba)
Returns:
Tensor containing the regularisation loss value.
"""
if self._fixed_mag:
if self._fixed_mag or self._fixed_prob: #Not enough DOF
return torch.tensor(0)
else:
#return reg_factor * F.l1_loss(self._params['mag'][self._reg_mask], target=self._reg_tgt, reduction='mean')
@ -559,7 +559,7 @@ class Data_augV7(nn.Module): #Proba sequentielles
Returns:
Tensor containing the regularisation loss value.
"""
if self._fixed_mag:
if self._fixed_mag or self._fixed_prob: #Not enough DOF
return torch.tensor(0)
else:
#return reg_factor * F.l1_loss(self._params['mag'][self._reg_mask], target=self._reg_tgt, reduction='mean')