search

facebookresearch / EGG

EGG: Emergence of lanGuage in Games
MIT License
286 stars 99 forks source link

fix callbacks #74

Closed robertodessi closed 4 years ago

robertodessi commented 4 years ago

fixing minor bugs in callbacks logics:

  1. early_stopper, even if it can store training stats, does not check them as a stopping criterion: I added a validation parameter to BaseEarlyStopper so that child classes can change that
  2. fix epoch counter when logging stats and when saving a model (both after loading one and when training from scratch)