From 8074745908fa86e642b91334ea749b061cff2c99 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 1 Aug 2020 19:24:14 -0700 Subject: [PATCH] hyperparameter evolution bug fix (#566) --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 26df910..9bd4d10 100644 --- 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))