File size: 2,909 Bytes
0453c63 | 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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 | # export PATH=/data/cuda/cuda-11.7/cuda/bin:$PATH
# export LD_LIBRARY_PATH=/data/cuda/cuda-11.7/cuda/lib64:$LD_LIBRARY_PATH
affordance_model="/gemini/code/AffordanceNet/ckpts/AffordanceVLM-7B"
# deepspeed --master_port=24990 train_aff.py \
# --version=$affordance_model \
# --dataset_dir='./data' \
# --dataset="reason_seg" \
# --sample_rates="1" \
# --vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
# --exp_name="AffordanceVLM-7B" \
# --eval_only \
# --eval_affordance \
# --val_dataset="handal_all"
# deepspeed --master_port=24991 train_aff.py \
# --version=$affordance_model \
# --dataset_dir='./data' \
# --dataset="reason_seg" \
# --sample_rates="1" \
# --vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
# --exp_name="AffordanceVLM-7B" \
# --eval_only \
# --eval_affordance \
# --val_dataset="handal_mini"
# deepspeed --master_port=24992 train_aff.py \
# --version=$affordance_model \
# --dataset_dir='/gemini/space/wrz/AffordanceNet/data' \
# --dataset="reason_seg" \
# --sample_rates="1" \
# --vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
# --exp_name="AffordanceVLM-7B" \
# --eval_only \
# --eval_affordance \
# --val_dataset="graspnet_test_seen"
# deepspeed --master_port=24993 train_aff.py \
# --version=$affordance_model \
# --dataset_dir='./data' \
# --dataset="reason_seg" \
# --sample_rates="1" \
# --vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
# --exp_name="AffordanceVLM-7B" \
# --eval_only \
# --eval_affordance \
# --val_dataset="graspnet_test_novel"
# deepspeed --master_port=24994 train_aff.py \
# --version=$affordance_model \
# --dataset_dir='./data' \
# --dataset="reason_seg" \
# --sample_rates="1" \
# --vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
# --exp_name="AffordanceVLM-7B" \
# --eval_only \
# --eval_affordance \
# --val_dataset="3doi"
# deepspeed --master_port=24995 train_aff.py \
# --version=$affordance_model \
# --dataset_dir='./data' \
# --dataset="reason_seg" \
# --sample_rates="1" \
# --vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
# --exp_name="AffordanceVLM-7B" \
# --eval_only \
# --eval_reason_aff \
# --val_dataset="handal_hard_reasoning"
deepspeed --master_port=24996 train_aff.py \
--version=$affordance_model \
--dataset_dir='/gemini/space/wrz/AffordanceNet/data' \
--dataset="reason_seg" \
--sample_rates="1" \
--vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
--exp_name="AffordanceVLM-7B" \
--eval_only \
--eval_reason_aff \
--val_dataset="handal_easy_reasoning"
# deepspeed --master_port=24997 train_aff.py \
# --version=$affordance_model \
# --dataset_dir='./data' \
# --dataset="reason_seg" \
# --sample_rates="1" \
# --vision_pretrained="ckpts/sam_vit_h_4b8939.pth" \
# --exp_name="AffordanceVLM-7B" \
# --eval_only \
# --eval_reason_aff \
# --val_dataset="3doi_easy_reasoning"
|