From 71c0ebcc365f98cdc07d98f3faa49dc6fec93ccb Mon Sep 17 00:00:00 2001 From: YifanXu Date: Fri, 20 Oct 2023 23:02:40 +0800 Subject: [PATCH] Update README.md --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 120cb89..3a6ca4f 100644 --- a/README.md +++ b/README.md @@ -177,6 +177,7 @@ Usually, we set python -m torch.distributed.launch --nproc_per_node=8 tools/train_net.py --config-file configs/pretrain/mq-glip-t.yaml --use-tensorboard OUTPUT_DIR 'OUTPUT/MQ-GLIP-TINY/' ``` To conduct pre-training, one should first extract vision queries before start training following the above [instruction](#vision-query-extraction). + To pre-train on custom datasets, please specify ``DATASETS.TRAIN`` and ``VISION_SUPPORT.SUPPORT_BANK_PATH`` in the config file. More details can be found in [CUSTOMIZED_PRETRAIN.md](CUSTOMIZED_PRETRAIN.md). ## Finetuning-free Evaluation