Update CUSTOMIZED_PRETRAIN.md

fix bug
BradyFU-patch-1
YifanXu 2023-10-23 15:42:32 +08:00 committed by GitHub
parent 0a26eb89b3
commit 8fc49fee2a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions

View File

@ -26,6 +26,7 @@ Here we use a new config file ``configs/pretrain/mq-glip-t_coco.yaml``.
python tools/train_net.py \
--config-file configs/pretrain/mq-glip-t_coco.yaml \
--extract_query \
VISION_QUERY.QUERY_BANK_PATH "" \
VISION_QUERY.QUERY_BANK_SAVE_PATH MODEL/coco_query_5000_sel_tiny.pth
```