char-rnn.pytorch icon indicating copy to clipboard operation
char-rnn.pytorch copied to clipboard

More modular approach and some novelties in params and training

Open zutotonno opened this issue 5 years ago • 0 comments

  • Its possible to add validation data and monitor overfitting by means of early stopping, as well as saving the model training at best checkpoints.
  • Its possible to launch an extensive grid search, but some more work need to be done.
  • Correctly handled the lstm module
  • Fixed minor bug during training sampling : if start index + chunk_len > len(file) -> exception
  • Allowing different type of training (random and non random sampling) and correctly 1 epoch -> 1 training set exploration.
  • Option to add dropout
  • Train/Validation evolution saved in .csv, as well as network configuration in json file

zutotonno avatar Jun 05 '19 11:06 zutotonno