 Bilal Khan
|
79526f82f5
|
Remove unnecessary epoch variable
|
2019-12-09 16:24:35 -05:00 |
|
|
Bilal Khan
|
9626e0458c
|
Add functionality to continue training from last saved global_step
|
2019-12-09 16:24:35 -05:00 |
|
|
Bilal Khan
|
2d73591a18
|
Stop saving current epoch
|
2019-12-09 16:24:35 -05:00 |
|
|
Bilal Khan
|
0eb973b0d9
|
Use saved optimizer and scheduler states if available
|
2019-12-09 16:24:35 -05:00 |
|
|
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 |
|