add arg

CEWu · Sep 13, 2022 · 4d9900d · 4d9900d
1 parent 463106b
commit 4d9900d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/upl_train.sh b/scripts/upl_train.sh
@@ -14,6 +14,7 @@ SHOTS=$5  # number of shots (1, 2, 4, 8, 16)
 CSC=$6  # class-specific context (False or True)
 CLASS_EQULE=$7  # CLASS_EQULE True of False
 TAG=$8 # log tag (multiple_models_random_init or rn50_random_init)
+FP=$9 # number of false positive training samples per class
 
 
 for SEED in {1..16}
@@ -30,6 +31,7 @@ do
         --dataset-config-file configs/datasets/${DATASET}.yaml \
         --config-file configs/trainers/${TRAINER}/${CFG}.yaml \
         --output-dir ${DIR} \
+        --num-fp ${FP} \
         TRAINER.UPLTrainer.N_CTX ${NCTX} \
         TRAINER.UPLTrainer.CSC ${CSC} \
         TRAINER.UPLTrainer.CLASS_TOKEN_POSITION ${CTP} \