cluster-rnn

commit 9532965f495522d60fef6243e80c3ae7e22ea625
parent cfcf0469a981484eb6c5cbb55576e350908f5728
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 14:10:16 -0500

mention rank in status updates

Diffstat:
M
train.lua
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.lua b/train.lua
@@ -427,7 +427,7 @@ for i = 1, iterations do
 
     if i % opt.print_every == 0 then
     machine_name = io.popen('hostname -s'):read()
-        print(string.format("%s %d/%d (epoch %.3f), train_loss = %6.8f, grad/param norm = %6.4e, time/batch = %.4fs",machine_name, i, iterations, epoch, train_loss, grad_params:norm() / params:norm(), time))
+        print(string.format("%s Rank %s %d/%d (epoch %.3f), train_loss = %6.8f, grad/param norm = %6.4e, time/batch = %.4fs",machine_name, rank, i, iterations, epoch, train_loss, grad_params:norm() / params:norm(), time))
     end
 
     if i % (opt.print_every*10) == 0 then