new version

cvfinetune/_version.py

@@ -1 +1 @@
-__version__ = "0.8.0"
+__version__ = "0.9.0"

cvfinetune/finetuner/mixins/trainer.py

@@ -69,13 +69,22 @@ class _TrainerMixin(BaseMixin):
             config=self.config,
             no_observe=self.no_observe)
 
-        # self.trainer will initialized later
+        # self.trainer will be initialized later
         def run(*args, **kwargs):
             self._check_attr("trainer")
             return self.trainer.run(*args, **kwargs)
 
         self.ex.main(run)
 
+    def run_experiment(self, *args, **kwargs):
+
+        if self.ex is None:
+            return self.trainer.run(*args, **kwargs)
+
+        sacred_reporter = SacredReport(ex=self.ex, trigger=(1, "epoch"))
+        self.trainer.extend(sacred_reporter)
+        return self.ex(*args, **kwargs)
+
 
     def init_updater(self):
         """Creates an updater from training iterator and the optimizer."""
@@ -164,16 +173,6 @@ class _TrainerMixin(BaseMixin):
             pyaml.dump(records, f, sort_keys=False)
 
 
-
-    def run_experiment(self, *args, **kwargs):
-
-        if self.ex is None:
-            return self.trainer.run(*args, **kwargs)
-
-        sacred_reporter = SacredReport(ex=self.ex, trigger=(1, "epoch"))
-        self.trainer.extend(sacred_reporter)
-        return self.ex(*args, **kwargs)
-
     def _trainer_output(self, name: str = ""):
         return Path(self.trainer.out, name)