Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Aug 2, 2020

Commit

8074745

1 Parent(s): e32abb5

hyperparameter evolution bug fix (#566)

Files changed (1) hide show

train.py CHANGED Viewed

@@ -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))

         # 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' % (yaml_file, yaml_file))