Add checkpoint logic - Githubissues

clay-lab / transductions

A PyTorch framework for creating, running, and reproducing experiments on seq2seq models.

6 stars 3 forks source link

Add checkpoint logic #38

Closed jopetty closed 3 years ago

jopetty commented 3 years ago

This should handle:

saving models
early stopping

jopetty commented 3 years ago

This is technically done. Right now, it's handled in the _load_checkpoint method of the Trainer, but it could be extracted to its own module to incorporate with early stopping and saving