BUG: fix two small mistakes with verbosity control (apache#168)

iblislin · Jan 15, 2017 · 030990a · 030990a
1 parent 1b09fb4
commit 030990a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)