Release of PromptSRC with pretrained models.
This commit is contained in:
39
configs/trainers/IVLP/vit_b16_c2_ep20_batch4_4+4ctx.yaml
Normal file
39
configs/trainers/IVLP/vit_b16_c2_ep20_batch4_4+4ctx.yaml
Normal file
@@ -0,0 +1,39 @@
|
||||
# Independent Vision Language Prompting
|
||||
DATALOADER:
|
||||
TRAIN_X:
|
||||
BATCH_SIZE: 4
|
||||
TEST:
|
||||
BATCH_SIZE: 100
|
||||
NUM_WORKERS: 8
|
||||
|
||||
INPUT:
|
||||
SIZE: (224, 224)
|
||||
INTERPOLATION: "bicubic"
|
||||
PIXEL_MEAN: [0.48145466, 0.4578275, 0.40821073]
|
||||
PIXEL_STD: [0.26862954, 0.26130258, 0.27577711]
|
||||
TRANSFORMS: ["random_resized_crop", "random_flip", "normalize"]
|
||||
|
||||
OPTIM:
|
||||
NAME: "sgd"
|
||||
LR: 0.0025
|
||||
MAX_EPOCH: 20
|
||||
LR_SCHEDULER: "cosine"
|
||||
WARMUP_EPOCH: 1
|
||||
WARMUP_TYPE: "constant"
|
||||
WARMUP_CONS_LR: 1e-5
|
||||
|
||||
TRAIN:
|
||||
PRINT_FREQ: 20
|
||||
|
||||
MODEL:
|
||||
BACKBONE:
|
||||
NAME: "ViT-B/16"
|
||||
|
||||
TRAINER:
|
||||
IVLP:
|
||||
N_CTX_VISION: 4
|
||||
N_CTX_TEXT: 4
|
||||
CTX_INIT: "a photo of a"
|
||||
PREC: "fp16"
|
||||
PROMPT_DEPTH_VISION: 9
|
||||
PROMPT_DEPTH_TEXT: 9
|
||||
Reference in New Issue
Block a user