minor fixes in the GC extension

cvfinetune/finetuner/mixins/trainer.py

@@ -19,9 +19,6 @@ from cvfinetune.training.extensions import SacredReport
 from cvfinetune.training.extensions import ManualGCCollect
 from cvfinetune.utils.sacred import Experiment
 
-@extension.make_extension(default_name="ManualGC", trigger=(1, "iteration"))
-def gc_collect(trainer):
-    gc.collect()
 
 class _TrainerMixin(BaseMixin):
     """This mixin is responsible for updater, evaluator and trainer creation.
@@ -132,7 +129,8 @@ class _TrainerMixin(BaseMixin):
         self.trainer = self._new_trainer(trainer_cls, opts, *args, **kwargs)
 
         if self.manual_gc:
-            self.trainer.extend(ManualGCCollect(trigger=(1, "iteration")))
+            manual_gc = ManualGCCollect(trigger=(1, "iteration"))
+            self.trainer.extend(manual_gc)
 
         self.save_meta_info()
 

cvfinetune/training/extensions/gc_collect.py

@@ -8,8 +8,6 @@ from chainer.training import trigger as trigger_module
 
 class ManualGCCollect(extension.Extension):
 
-	# should one of the last extensions
-	priority = extension.PRIORITY_WRITER + 1
 	SLEEP = 3
 
 	def __init__(self, trigger=(1, "iteration")):