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

Higher patch

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-04 11:56:45 -05:00
parent be8491268a
commit a26252feea
3 changed files with 31 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
higher/smart_aug/dataug.py
View file

@ -19,6 +19,12 @@ import copy
import transformations as TF
import higher
import higher_patch
from utils import clip_norm
from train_utils import compute_vaLoss
### Data augmenter ###
class Data_augV5(nn.Module): #Optimisation jointe (mag, proba)
"""Data augmentation module with learnable parameters.
@ -798,7 +804,6 @@ class RandAug(nn.Module): #RandAugment = UniformFx-MagFxSh + rapide
return "RandAug(%dTFx%d-Mag%d)" % (self._nb_tf, self._N_seqTF, self.mag)
### Models ###
import higher
class Higher_model(nn.Module):
"""Model wrapper for higher gradient tracking.
@ -897,8 +902,6 @@ class Higher_model(nn.Module):
"""
return self._name
from utils import clip_norm
from train_utils import compute_vaLoss
class Augmented_model(nn.Module):
"""Wrapper for a Data Augmentation module and a model.