search

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
https://detrex.readthedocs.io/en/latest/
Apache License 2.0
1.95k stars 204 forks source link

Is checkpointing supported / should it be added? #228

Closed FabianSchuetze closed 1 year ago

FabianSchuetze commented 1 year ago

Thanks for the wonderful repo - it's a pleasure to work with it and to read the code.

Is checkpointing currently supported by detrex? If not, would there be an interest in a PR that adds checkpointing? I have added it for the Dino encoder and decoder and it seemed to work well.

rentainhe commented 1 year ago

Thanks for the wonderful repo - it's a pleasure to work with it and to read the code.

Is checkpointing currently supported by detrex? If not, would there be an interest in a PR that adds checkpointing? I have added it for the Dino encoder and decoder and it seemed to work well.

Sure, we appreciate any kind of contribution from the community~ You can open a pull request about this, thank you so much~

rentainhe commented 1 year ago

BTW, we simply support checkpointing in DINO before, but a little bit not that elegant, we will highly appreciate your pull request about it~

FabianSchuetze commented 1 year ago

I just saw the new release and PR #200 . The solution is very elegant and useful. Thanks a lot!