Browse Source

train.py: printing correct variable

master
Rafael Valle 4 years ago
parent
commit
6d0635e8c1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      train.py

+ 1
- 1
train.py View File

@ -142,7 +142,7 @@ def validate(model, criterion, valset, iteration, batch_size, n_gpus,
model.train() model.train()
if rank == 0: if rank == 0:
print("Validation loss {}: {:9f} ".format(iteration, reduced_val_loss))
print("Validation loss {}: {:9f} ".format(iteration, val_loss))
logger.log_validation(val_loss, model, y, y_pred, iteration) logger.log_validation(val_loss, model, y, y_pred, iteration)

Loading…
Cancel
Save