How can I save checkpoint?

[teang1995]

Hi, how can i save checkpoint while training retrieval model? I cannot find where to save model checkpoint.

Thank you.

[simon-ging]

hi, checkpoints are saved automatically, check the experiments folder and also the documentation here: https://github.com/gingsi/coot-videotext/tree/master/nntrainer

The method that saves the checkpoints is _save_checkpoint in class BaseTrainer file nntrainer/trainer_base.py