Browse Source

new version

Dimitri Korsch 3 years ago
parent
commit
aaf67291ab
2 changed files with 5 additions and 3 deletions
  1. 1 1
      cvfinetune/_version.py
  2. 4 2
      cvfinetune/training/trainer/base.py

+ 1 - 1
cvfinetune/_version.py

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

+ 4 - 2
cvfinetune/training/trainer/base.py

@@ -40,12 +40,14 @@ class Trainer(T):
 		updater,
 		evaluator: extensions.Evaluator = None,
 		intervals: attr_dict = default_intervals,
-		no_observe: bool = False):
+		no_observe: bool = False,
+		**kwargs):
 
 		super(Trainer, self).__init__(
 			updater=updater,
 			stop_trigger=(opts.epochs, 'epoch'),
-			out=opts.output
+			out=opts.output,
+			**kwargs
 		)
 		logging.info("Training outputs are saved under \"{}\"".format(self.out))