search

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.71k stars 307 forks source link

Save checkpoint after certain number of iterations #227

Closed sparshgarg23 closed 2 years ago

sparshgarg23 commented 2 years ago

Currently checkpoint is being saved after 1 epoch.Since I am training on colab at a time I can only use one gpu which is causing the training to proceed very slowly.

With that in mind,what changes should I make to the config so as to save checkpoint after every 500 iterations?

sparshgarg23 commented 2 years ago

I was able to discuss this at mmdetection. it seems the above idea of saving checkpoint after N iterations can't be done because the mmcv version of this repository doesn't support by_epoch=False.

As such would it be possible to update the repository so that it's consistent with the current mmcv pytorch and mmdetection standard?