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

@@ -536,4 +536,4 @@ if __name__ == '__main__':
# Plot results
plot_evolution_results(yaml_file)
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