Procházet zdrojové kódy

More informative `EarlyStopping()` message (#5303)

modifyDataloader
Glenn Jocher GitHub před 2 roky
rodič
revize
441b47c443
V databázi nebyl nalezen žádný známý klíč pro tento podpis ID GPG klíče: 4AEE18F83AFDEB23
1 změnil soubory, kde provedl 4 přidání a 1 odebrání
  1. +4
    -1
      utils/torch_utils.py

+ 4
- 1
utils/torch_utils.py Zobrazit soubor

@@ -299,7 +299,10 @@ class EarlyStopping:
self.possible_stop = delta >= (self.patience - 1) # possible stop may occur next epoch
stop = delta >= self.patience # stop training if patience exceeded
if stop:
LOGGER.info(f'EarlyStopping patience {self.patience} exceeded, stopping training.')
LOGGER.info(f'Stopping training early as no improvement observed in last {self.patience} epochs. '
f'Best results observed at epoch {self.best_epoch}, best model saved as best.pt.\n'
f'To update EarlyStopping(patience={self.patience}) pass a new patience value, '
f'i.e. `python train.py --patience 300` or use `--patience 0` to disable EarlyStopping.')
return stop



Načítá se…
Zrušit
Uložit