[Bug] Update checkpoint loading func

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.97k stars 206 forks source link

Closed rentainhe closed 1 year ago

rentainhe commented 1 year ago

To use this updated checkpointer, you should modify tools/train_net.py by:

# from detectron2.checkpoint import DetectionCheckpointer
from detrex.checkpoint import DetectionCheckpointer

We did not set this as the default one.