From 70d37f9e7d4a74ba4169b91114e936b446f79893 Mon Sep 17 00:00:00 2001 From: Rafael Valle Date: Fri, 25 Oct 2019 21:45:37 -0700 Subject: [PATCH] train.py: reporting the right variable --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index e93917b..61f0888 100644 --- a/train.py +++ b/train.py @@ -143,7 +143,7 @@ def validate(model, criterion, valset, iteration, batch_size, n_gpus, model.train() if rank == 0: print("Validation loss {}: {:9f} ".format(iteration, reduced_val_loss)) - logger.log_validation(reduced_val_loss, model, y, y_pred, iteration) + logger.log_validation(val_loss, model, y, y_pred, iteration) def train(output_directory, log_directory, checkpoint_path, warm_start, n_gpus,