From 0df94e5b47c9984e48069a6b7f98b6d59d737f76 Mon Sep 17 00:00:00 2001 From: YifanXu Date: Fri, 20 Oct 2023 22:54:10 +0800 Subject: [PATCH] Update CUSTOMIZED_PRETRAIN.md --- CUSTOMIZED_PRETRAIN.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/CUSTOMIZED_PRETRAIN.md b/CUSTOMIZED_PRETRAIN.md index 70d1a01..fe1e160 100644 --- a/CUSTOMIZED_PRETRAIN.md +++ b/CUSTOMIZED_PRETRAIN.md @@ -14,7 +14,7 @@ Here we add a new dataset ``coco_grounding_train_for_obj365``. **2. Acquire customized config files** -You can modified upon the [official pretraining config file](configs/pretrain/mq-glip-t.yaml) to get a customized config file. Here we provide an [example](configs/pretrain/mq-glip-t_coco.yaml). You customize your own needs following the "``NOTE``" in the file. +You can modified upon the [official pretraining config file](configs/pretrain/mq-glip-t.yaml) to get a customized config file. Here we provide an [example](configs/pretrain/mq-glip-t_coco.yaml). You can customize your own needs following the "``NOTE``" in the file. Make sure to use correct ``DATASETS.TRAIN`` and ``VISION_QUERY.QUERY_BANK_PATH``. @@ -37,4 +37,4 @@ You can specify ``VISION_QUERY.MAX_QUERY_NUMBER`` (number of queries for each ca **4. Conduct modulated pretraining** ``` python -m torch.distributed.launch --nproc_per_node=8 tools/train_net.py --config-file configs/pretrain/mq-glip-t_coco.yaml --use-tensorboard OUTPUT_DIR 'OUTPUT/MQ-GLIP-TINY-COCO/' -``` \ No newline at end of file +```