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