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