Commit b78567bb authored by Vít Starý Novotný's avatar Vít Starý Novotný
Browse files

Prune old checkpoints

parent 72d22741

ahisto_named_entity_search/recognition/model.py

+1 −0
Original line number Diff line number Diff line
@@ -93,6 +93,7 @@ class NerModel: 
            stopping_patience=cls.STOPPING_PATIENCE,

            save_strategy='steps',

            save_steps=cls.SAVE_STEPS,

            save_total_limit=cls.STOPPING_PATIENCE + 1,

            logging_strategy='steps',

            logging_steps=cls.LOGGING_STEPS,

            do_train=True,