|
@@ -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))
|
|
|
|