Parcourir la source

Update train.py

modifyDataloader
Glenn Jocher GitHub il y a 2 ans
Parent
révision
a5c9057dcc
Aucune clé connue n'a été trouvée dans la base pour cette signature ID de la clé GPG: 4AEE18F83AFDEB23
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. +2
    -2
      train.py

+ 2
- 2
train.py Voir le fichier

@@ -623,9 +623,9 @@ def main(opt, callbacks=Callbacks()):

# Plot results
plot_evolve(evolve_csv)
LOGGER.info(f'Hyperparameter evolution finished\n'
LOGGER.info(f'Hyperparameter evolution finished {opt.evolve} generations\n'
f"Results saved to {colorstr('bold', save_dir)}\n"
f'Use best hyperparameters example: $ python train.py --hyp {evolve_yaml}')
f'Usage example: $ python train.py --hyp {evolve_yaml}')


def run(**kwargs):

Chargement…
Annuler
Enregistrer