hyperparameter evolution bug fix (#566)

5 years ago · 8074745908
parent e32abb5fb9
commit 8074745908
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -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))