Release of PromptSRC with pretrained models.
This commit is contained in:
17
scripts/zsclip/zeroshot.sh
Normal file
17
scripts/zsclip/zeroshot.sh
Normal file
@@ -0,0 +1,17 @@
|
||||
#!/bin/bash
|
||||
|
||||
#cd ../..
|
||||
|
||||
# custom config
|
||||
DATA=/path/to/datasets
|
||||
TRAINER=ZeroshotCLIP
|
||||
DATASET=$1
|
||||
CFG=$2 # rn50, rn101, vit_b32 or vit_b16
|
||||
|
||||
python train.py \
|
||||
--root ${DATA} \
|
||||
--trainer ${TRAINER} \
|
||||
--dataset-config-file configs/datasets/${DATASET}.yaml \
|
||||
--config-file configs/trainers/CoOp/${CFG}.yaml \
|
||||
--output-dir output/${TRAINER}/${CFG}/${DATASET} \
|
||||
--eval-only
|
||||
Reference in New Issue
Block a user