File size: 1,129 Bytes
9e14838 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | #!/usr/bin/env bash
# clip svd
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
CUDA_VISIBLE_DEVICES=1 python train.py \
--name clip_vitl14_svd \
--wang2020_data_path /Youtu_Pangu_Security_Public/etoilefu/CNNDetection/dataset/ \
--data_mode wang2020 \
--arch CLIP:ViT-L/14_svd \
--batch_size 48 \
--loadSize 256 \
--cropSize 224 \
--lr 0.0002 \
--use_svd
# siglip svd
# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
# CUDA_VISIBLE_DEVICES=1 python train.py \
# --name siglip_vitl14_svd \
# --wang2020_data_path /Youtu_Pangu_Security_Public/etoilefu/CNNDetection/dataset/ \
# --data_mode wang2020 \
# --arch SigLIP:ViT-L/16_256_svd \
# --batch_size 48 \
# --loadSize 256 \
# --cropSize 256 \
# --lr 0.0002 \
# --use_svd
# beitv2 svd
# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
# CUDA_VISIBLE_DEVICES=1 python train.py \
# --name beitv2_vitl14_svd \
# --wang2020_data_path /Youtu_Pangu_Security_Public/etoilefu/CNNDetection/dataset/ \
# --data_mode wang2020 \
# --arch BEiTv2:ViT-L/16_svd \
# --batch_size 48 \
# --loadSize 256 \
# --cropSize 224 \
# --lr 0.0002 \
# --use_svd |