コード例 #1
0
ファイル: _freeze.py プロジェクト: MIPT-Oulu/Collagen 
 def on_batch_end(self, *args, **kwargs):
     freeze_modules(self.__modules, invert=True)
コード例 #2
0
ファイル: _freeze.py プロジェクト: MIPT-Oulu/Collagen 
 def on_sample_end(self, *args, **kwargs):
     freeze_modules(self.__modules, invert=True)
コード例 #3
0
ファイル: _freeze.py プロジェクト: MIPT-Oulu/Collagen 
 def on_batch_begin(self, *args, **kwargs):
     freeze_modules(self.__modules)
コード例 #4
0
ファイル: _freeze.py プロジェクト: MIPT-Oulu/Collagen 
 def on_sample_begin(self, *args, **kwargs):
     freeze_modules(self.__modules)