Browse Source

hyperparameter evolution bug fix (#566)

5.0
Glenn Jocher 4 years ago
parent
commit
8074745908
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      train.py

+ 1
- 1
train.py View File

# Plot results # Plot results
plot_evolution_results(yaml_file) plot_evolution_results(yaml_file)
print('Hyperparameter evolution complete. Best results saved as: %s\nCommand to train a new model with these ' print('Hyperparameter evolution complete. Best results saved as: %s\nCommand to train a new model with these '
'hyperparameters: $ python train.py --hyp %s' % (f, f))
'hyperparameters: $ python train.py --hyp %s' % (yaml_file, yaml_file))

Loading…
Cancel
Save