From 030990a155fab5c1e36075800f144f72c3424901 Mon Sep 17 00:00:00 2001 From: Spencer Lyon Date: Sun, 15 Jan 2017 08:38:44 -0500 Subject: [PATCH] BUG: fix two small mistakes with verbosity control (#168) --- src/model.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/model.jl b/src/model.jl index 98a1a21a216f..3b38cf3193f2 100644 --- a/src/model.jl +++ b/src/model.jl @@ -522,7 +522,7 @@ function fit(self :: FeedForward, optimizer :: AbstractOptimizer, data :: Abstra time_stop = time() metric = get(opts.eval_metric) - opts.verbosity >= 2 && info(format("== Epoch {1:0>3d}/{1:0>3d} ==========", i_epoch, opts.n_epoch)) + opts.verbosity >= 2 && info(format("== Epoch {1:0>3d}/{2:0>3d} ==========", i_epoch, opts.n_epoch)) if opts.verbosity >= 3 info("## Training summary") for (name, value) in metric @@ -579,6 +579,7 @@ function fit(self :: FeedForward, optimizer :: AbstractOptimizer, data :: Abstra end # end of all epochs opts.verbosity >= 1 && info("Finish training on $(self.ctx)") + nothing end function save_checkpoint(self :: FeedForward, prefix :: AbstractString, state :: OptimizationState)