From 3e63ba2d17bf8f6d1fc3c3cd1d00be991e38898b Mon Sep 17 00:00:00 2001 From: Colombo Date: Thu, 30 Jan 2020 07:49:45 +0400 Subject: [PATCH] fix mean loss history after save --- mainscripts/Trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index b8d62d3..e137ff0 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -133,7 +133,7 @@ def trainerThread (s2c, c2s, e, shared_state['after_save'] = False last_save_time = time.time() - mean_loss = np.mean ( [ np.array(loss_history[i]) for i in range(save_iter, iter) ], axis=0) + mean_loss = np.mean ( loss_history[save_iter:iter], axis=0) for loss_value in mean_loss: loss_string += "[%.4f]" % (loss_value)