Python freeze_modules Examples

Example #1

File: _freeze.py Project: MIPT-Oulu/Collagen

 def on_batch_end(self, *args, **kwargs):
     freeze_modules(self.__modules, invert=True)

Example #2

File: _freeze.py Project: MIPT-Oulu/Collagen

 def on_sample_end(self, *args, **kwargs):
     freeze_modules(self.__modules, invert=True)

Example #3

File: _freeze.py Project: MIPT-Oulu/Collagen

 def on_batch_begin(self, *args, **kwargs):
     freeze_modules(self.__modules)

Example #4

File: _freeze.py Project: MIPT-Oulu/Collagen

 def on_sample_begin(self, *args, **kwargs):
     freeze_modules(self.__modules)