4 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Bilal Khan	a03fcf570d	Save tokenizer after each epoch to be able to resume training from a checkpoint	2019-12-09 16:24:35 -05:00
Bilal Khan	f71b1bb05a	Save optimizer state, scheduler state and current epoch	2019-12-09 16:24:35 -05:00
Bilal Khan	07bf43074f	Fix GPT2 docstring	2019-11-25 11:32:00 -05:00
Bilal Khan	5ce8d29abe	Change tensorboard imports to use built-in tensorboard if available	2019-10-08 16:29:43 -05:00