Browse Source

train.py: reporting the right variable

master
Rafael Valle 5 years ago
committed by GitHub
parent
commit
70d37f9e7d
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      train.py

+ 1
- 1
train.py View File

@ -143,7 +143,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, 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, def train(output_directory, log_directory, checkpoint_path, warm_start, n_gpus,

Loading…
Cancel
Save