File size: 547 Bytes
1633fcc
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12


#### access DINOv2
export CUDA_VISIBLE_DEVICES=3;
python train_decode_proteus_feats.py \
    --batch-size 64 --warmup-epochs 5 --epochs 300 \
    --data-set IMNET --data-path '' \
    --teacher-model vit_large --target_model vit_base --model models_proteus_dinov2 \
    --patch_size 14 --mask_probability 0.5 --mask_ratio 0.5 --mask_first_n \
    --lambda_token 1.0 --lambda_fea 1.0 --lambda_patch 1.0 \
    --finetune "/home/t2vg-a100-G4-1/projects/qiyp/1_feature_extractor/1_feature_extractor/checkpoint0160.pth" \
    --output_dir log/IF_vis;