google-research / electra

ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators
Apache License 2.0
2.33k stars 352 forks source link

Add keep_checkpoint_max parameter #47

Closed stefan-it closed 4 years ago

stefan-it commented 4 years ago

Hi,

this PR adds keep_checkpoint_max as new parameter to the RunConfig instance, in order to allow the user to specify the max. number of recent checkpoints to keep.

By default, only 5 checkpoints are kept. But this number can be quite low when evaluating own trained models, so the number of checkpoints to keep can be set now.