diff --git a/configs/pretrain/mq-groundingdino-t.yaml b/configs/pretrain/mq-groundingdino-t.yaml index a535339..7a521d1 100644 --- a/configs/pretrain/mq-groundingdino-t.yaml +++ b/configs/pretrain/mq-groundingdino-t.yaml @@ -54,6 +54,7 @@ AUGMENT: DATALOADER: SIZE_DIVISIBILITY: 32 + NUM_WORKERS: 0 SOLVER: OPTIMIZER: ADAMW @@ -75,7 +76,7 @@ SOLVER: MODEL_EMA: 0.999 FIND_UNUSED_PARAMETERS: False CHECKPOINT_PERIOD: 99999999 - CHECKPOINT_PER_EPOCH: 32.0 + CHECKPOINT_PER_EPOCH: 2.0 # TUNING_HIGHLEVEL_OVERRIDE: "vision_query" TUNING_HIGHLEVEL_OVERRIDE: "vision_query" MAX_TO_KEEP: 4