search

open-mmlab / mmselfsup

OpenMMLab Self-Supervised Learning Toolbox and Benchmark
https://mmselfsup.readthedocs.io/en/latest/
Apache License 2.0
3.14k stars 429 forks source link

[Docs] Training Object detector using MAE as backbone #777

Open sandesha-hegde opened 5 months ago

sandesha-hegde commented 5 months ago

The doc issue

I only see the below commands but don't have any examples to train it on other pretrained models like MAE

bash tools/benchmarks/mmdetection/mim_dist_train_c4.sh ${CONFIG} ${PRETRAIN} ${GPUS}
bash tools/benchmarks/mmdetection/mim_dist_train_fpn.sh ${CONFIG} ${PRETRAIN} ${GPUS}

# slurm version
bash tools/benchmarks/mmdetection/mim_slurm_train_c4.sh ${PARTITION} ${CONFIG} ${PRETRAIN}
bash tools/benchmarks/mmdetection/mim_slurm_train_fpn.sh ${PARTITION} ${CONFIG} ${PRETRAIN}

Can you please provide the more details or some sample MAE config files so that it will help us to train the object detectors.

Suggest a potential alternative/fix

No response