glenn-jocher commited on
Commit
8074745
1 Parent(s): e32abb5

hyperparameter evolution bug fix (#566)

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -536,4 +536,4 @@ if __name__ == '__main__':
536
  # Plot results
537
  plot_evolution_results(yaml_file)
538
  print('Hyperparameter evolution complete. Best results saved as: %s\nCommand to train a new model with these '
539
- 'hyperparameters: $ python train.py --hyp %s' % (f, f))
 
536
  # Plot results
537
  plot_evolution_results(yaml_file)
538
  print('Hyperparameter evolution complete. Best results saved as: %s\nCommand to train a new model with these '
539
+ 'hyperparameters: $ python train.py --hyp %s' % (yaml_file, yaml_file))