Ver código fonte

hyperparameter evolution bug fix (#566)

5.0
Glenn Jocher 4 anos atrás
pai
commit
8074745908
1 arquivos alterados com 1 adições e 1 exclusões
  1. +1
    -1
      train.py

+ 1
- 1
train.py Ver arquivo

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

Carregando…
Cancelar
Salvar