Esempio n. 1
0
 def on_batch_end(self, *args, **kwargs):
     freeze_modules(self.__modules, invert=True)
Esempio n. 2
0
 def on_sample_end(self, *args, **kwargs):
     freeze_modules(self.__modules, invert=True)
Esempio n. 3
0
 def on_batch_begin(self, *args, **kwargs):
     freeze_modules(self.__modules)
Esempio n. 4
0
 def on_sample_begin(self, *args, **kwargs):
     freeze_modules(self.__modules)