Detectron2: Does the model save automatically during training, if no where to add that?

Created on 8 Nov 2019  ยท  7Comments  ยท  Source: facebookresearch/detectron2

โ“ Questions and Help

Does the model save automatically when being trained or do we have to add that? If we have to add it where would be a good place to do it if we want to save for every epoch instead of at the very end of the training?

Most helpful comment

controlled by config SOLVER.CHECKPOINT_PERIOD.

All 7 comments

controlled by config SOLVER.CHECKPOINT_PERIOD.

How to save the model in .pth format? If using cfg.OUTPUT_DIR is the answer, why can I not find the output in the current directory I use?

It is saved every SOLVER.CHECKPOINT_PERIOD iterations.

It is saved every SOLVER.CHECKPOINT_PERIOD iterations.

Sorry I cannot find the documentation in the documentation website. How to export it into model.pth format?

The model will be saved to .pth format during training after every SOLVER.CHECKPOINT_PERIOD iterations.

Can we rename the model_final.pth ?

You cannot choose the name. You can rename it after it is saved.

Was this page helpful?
0 / 5 - 0 ratings

Related issues

invisprints picture invisprints  ยท  4Comments

AntonBaumannDE picture AntonBaumannDE  ยท  3Comments

ChungNPH picture ChungNPH  ยท  3Comments

Cold-Winter picture Cold-Winter  ยท  3Comments

RomRoc picture RomRoc  ยท  4Comments