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

Modification test pour simplifier early stopping

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-13 13:38:00 -05:00
parent 198fb06065
commit f0c0559e73
4 changed files with 49 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

3
higher/dataug.py
View file

@ -553,9 +553,10 @@ class Augmented_model(nn.Module):
mode=self._data_augmentation
self._mods['data_aug'].augment(mode)
super(Augmented_model, self).train(mode)
return self
def eval(self):
self.train(mode=False)
return self.train(mode=False)
#super(Augmented_model, self).eval()
def items(self):