Qiyp's picture
code of stage1 & 3, remove large files
1633fcc
#### 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;