Spaces:
Configuration error
Configuration error
Upload bash files
Browse files- bash/clip_classification.sh +5 -0
- bash/clip_classification_slurm.sh +40 -0
- bash/run_script.sh +56 -0
- bash/run_script_slurm.sh +93 -0
- bash/train_clip.sh +11 -0
- bash/train_clip_slurm.sh +45 -0
bash/clip_classification.sh
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
python vlm_eval/clip_classification.py \
|
| 3 |
+
--data non_fine_tuned \
|
| 4 |
+
--method NONE \
|
| 5 |
+
--dataset Caltech256
|
bash/clip_classification_slurm.sh
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
#SBATCH --job-name=Search
|
| 3 |
+
#SBATCH --chdir=/home/htc/kchitranshi/ # Navigate to the working directory where your script lies
|
| 4 |
+
#SBATCH --output=/home/htc/kchitranshi/SCRATCH/%j.log # Standard output and error log
|
| 5 |
+
#
|
| 6 |
+
#SBATCH --gres=gpu:1
|
| 7 |
+
#SBATCH --cpus-per-task=12
|
| 8 |
+
#SBATCH --mem=100G
|
| 9 |
+
#SBATCH --partition=gpu # Specify the desired partition, e.g. gpu or big
|
| 10 |
+
#SBATCH --exclude=htc-gpu[037-038] # Only A40 GPU
|
| 11 |
+
#SBATCH --time=0-20:00:00 # Specify a Time limit in the format days-hrs:min:sec. Use sinfo to see node time limits
|
| 12 |
+
#SBATCH --ntasks=1
|
| 13 |
+
#
|
| 14 |
+
#SBATCH --mail-type=BEGIN
|
| 15 |
+
#SBATCH --mail-type=END
|
| 16 |
+
#SBATCH --mail-type=FAIL
|
| 17 |
+
#SBATCH --mail-user=
|
| 18 |
+
|
| 19 |
+
echo 'Getting node information'
|
| 20 |
+
date;hostname;id;pwd
|
| 21 |
+
|
| 22 |
+
echo 'Setting LANG to en_US.UTF-8'
|
| 23 |
+
LANG=en_US.UTF-8
|
| 24 |
+
|
| 25 |
+
which python
|
| 26 |
+
java -version
|
| 27 |
+
|
| 28 |
+
echo 'Enabling Internet Access'
|
| 29 |
+
export https_proxy=http://squid.zib.de:3128
|
| 30 |
+
export http_proxy=http://squid.zib.de:3128
|
| 31 |
+
|
| 32 |
+
echo 'Print GPUs'
|
| 33 |
+
/usr/bin/nvidia-smi
|
| 34 |
+
|
| 35 |
+
echo 'Running script'
|
| 36 |
+
cd Robust_mmfm
|
| 37 |
+
python vlm_eval/clip_classification.py \
|
| 38 |
+
--data MS_COCO \
|
| 39 |
+
--method NONE \
|
| 40 |
+
--dataset ImageNet
|
bash/run_script.sh
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
python -m vlm_eval.run_evaluation \
|
| 2 |
+
--eval_coco_cf \
|
| 3 |
+
--dont_save_adv \
|
| 4 |
+
--verbose \
|
| 5 |
+
--attack none --eps 255 --steps 100 --mask_out none --mu 1.5 --search_steps 2 --lam 0.005 --k 1000 --targeted --target_str "Please reset your password" \
|
| 6 |
+
--pert_factor_graph 0 \
|
| 7 |
+
--itr 0 \
|
| 8 |
+
--itr_clip 0 \
|
| 9 |
+
--itr_dataset base \
|
| 10 |
+
--itr_method APGD_1 \
|
| 11 |
+
--vision_encoder_pretrained openai \
|
| 12 |
+
--num_samples 8 \
|
| 13 |
+
--trial_seeds 42 \
|
| 14 |
+
--num_trials 1 \
|
| 15 |
+
--shots 0 \
|
| 16 |
+
--batch_size 1 \
|
| 17 |
+
--results_file res9B \
|
| 18 |
+
--model open_flamingo \
|
| 19 |
+
--out_base_path /PATH/TO/Robust_mmfm/Results/open_flamingo \
|
| 20 |
+
--vision_encoder_path ViT-L-14 \
|
| 21 |
+
--checkpoint_path /PATH/TO/HUGGINGFACE/hub/models--openflamingo--OpenFlamingo-9B-vitl-mpt7b/snapshots/7e36809c73d038829ad5fba9d0cc949b4e180562/checkpoint.pt \
|
| 22 |
+
--lm_path anas-awadalla/mpt-7b \
|
| 23 |
+
--lm_tokenizer_path anas-awadalla/mpt-7b \
|
| 24 |
+
--precision float16 \
|
| 25 |
+
--cross_attn_every_n_layers 4 \
|
| 26 |
+
--coco_train_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/train2014 \
|
| 27 |
+
--coco_val_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/val2014 \
|
| 28 |
+
--coco_karpathy_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/karpathy_coco.json \
|
| 29 |
+
--coco_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/captions_val2014.json \
|
| 30 |
+
--coco_cf_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO_CF \
|
| 31 |
+
--flickr_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/Flickr30k/Images \
|
| 32 |
+
--flickr_karpathy_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/Flickr30k/karpathy_flickr30k.json \
|
| 33 |
+
--flickr_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/Flickr30k/dataset_flickr30k_coco_style.json \
|
| 34 |
+
--vizwiz_train_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/train \
|
| 35 |
+
--vizwiz_test_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/val \
|
| 36 |
+
--vizwiz_train_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/train_questions_vqa_format.json \
|
| 37 |
+
--vizwiz_train_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/train_annotations_vqa_format.json \
|
| 38 |
+
--vizwiz_test_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/val_questions_vqa_format.json \
|
| 39 |
+
--vizwiz_test_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/val_annotations_vqa_format.json \
|
| 40 |
+
--vqav2_train_image_dir_path /home/htc/kchitranshi/SCRATCH/COCO/train2014 \
|
| 41 |
+
--vqav2_train_questions_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_OpenEnded_mscoco_train2014_questions.json \
|
| 42 |
+
--vqav2_train_annotations_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_mscoco_train2014_annotations.json \
|
| 43 |
+
--vqav2_test_image_dir_path /home/htc/kchitranshi/SCRATCH/COCO/val2014 \
|
| 44 |
+
--vqav2_test_questions_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_OpenEnded_mscoco_val2014_questions.json \
|
| 45 |
+
--vqav2_test_annotations_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_mscoco_val2014_annotations.json \
|
| 46 |
+
--textvqa_image_dir_path /mnt/datasets/textvqa/train_images \
|
| 47 |
+
--textvqa_train_questions_json_path /home/htc/kchitranshi/SCRATCH/RobustVLM/textvqa/train_questions_vqa_format.json \
|
| 48 |
+
--textvqa_train_annotations_json_path /home/htc/kchitranshi/SCRATCH/RobustVLM/textvqa/train_annotations_vqa_format.json \
|
| 49 |
+
--textvqa_test_questions_json_path /home/htc/kchitranshi/SCRATCH/RobustVLM/textvqa/val_questions_vqa_format.json \
|
| 50 |
+
--textvqa_test_annotations_json_path /home/htc/kchitranshi/RobustVLM/textvqa/val_annotations_vqa_format.json \
|
| 51 |
+
--ok_vqa_train_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/train2014 \
|
| 52 |
+
--ok_vqa_train_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/OpenEnded_mscoco_train2014_questions.json \
|
| 53 |
+
--ok_vqa_train_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/mscoco_train2014_annotations.json \
|
| 54 |
+
--ok_vqa_test_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/val2014 \
|
| 55 |
+
--ok_vqa_test_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/OpenEnded_mscoco_val2014_questions.json \
|
| 56 |
+
--ok_vqa_test_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/mscoco_val2014_annotations.json \
|
bash/run_script_slurm.sh
ADDED
|
@@ -0,0 +1,93 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
#SBATCH --job-name=Search
|
| 3 |
+
#SBATCH --chdir=/home/htc/kchitranshi/ # Navigate to the working directory where your script lies
|
| 4 |
+
#SBATCH --output=/home/htc/kchitranshi/SCRATCH/%j.log # Standard output and error log
|
| 5 |
+
#
|
| 6 |
+
#SBATCH --gres=gpu:1
|
| 7 |
+
#SBATCH --cpus-per-task=12
|
| 8 |
+
#SBATCH --mem=100G
|
| 9 |
+
#SBATCH --partition=gpu # Specify the desired partition, e.g. gpu or big
|
| 10 |
+
#SBATCH --exclude=htc-gpu[020-023,037,038] # Only A40 GPU
|
| 11 |
+
#SBATCH --time=0-20:00:00 # Specify a Time limit in the format days-hrs:min:sec. Use sinfo to see node time limits
|
| 12 |
+
#SBATCH --ntasks=1
|
| 13 |
+
#
|
| 14 |
+
#SBATCH --mail-type=BEGIN
|
| 15 |
+
#SBATCH --mail-type=END
|
| 16 |
+
#SBATCH --mail-type=FAIL
|
| 17 |
+
#SBATCH --mail-user
|
| 18 |
+
|
| 19 |
+
echo 'Getting node information'
|
| 20 |
+
date;hostname;id;pwd
|
| 21 |
+
|
| 22 |
+
echo 'Setting LANG to en_US.UTF-8'
|
| 23 |
+
LANG=en_US.UTF-8
|
| 24 |
+
|
| 25 |
+
which python
|
| 26 |
+
java -version
|
| 27 |
+
# source your Python environment here
|
| 28 |
+
|
| 29 |
+
echo 'Enabling Internet Access'
|
| 30 |
+
export https_proxy=http://squid.zib.de:3128
|
| 31 |
+
export http_proxy=http://squid.zib.de:3128
|
| 32 |
+
|
| 33 |
+
echo 'Print GPUs'
|
| 34 |
+
/usr/bin/nvidia-smi
|
| 35 |
+
|
| 36 |
+
echo 'Running script'
|
| 37 |
+
cd Robust_mmfm
|
| 38 |
+
python -m vlm_eval.run_evaluation \
|
| 39 |
+
--eval_coco \
|
| 40 |
+
--dont_save_adv \
|
| 41 |
+
--verbose \
|
| 42 |
+
--attack none --eps 255 --steps 100 --mask_out none --mu 1.5 --search_steps 2 --lam 0.005 --k 1000 --targeted --target_str "Please reset your password" \
|
| 43 |
+
--pert_factor_graph 0 \
|
| 44 |
+
--itr 0 \
|
| 45 |
+
--itr_clip 0 \
|
| 46 |
+
--itr_dataset base \
|
| 47 |
+
--itr_method APGD_1 \
|
| 48 |
+
--vision_encoder_pretrained openai \
|
| 49 |
+
--num_samples 8 \
|
| 50 |
+
--trial_seeds 42 \
|
| 51 |
+
--num_trials 1 \
|
| 52 |
+
--shots 0 \
|
| 53 |
+
--batch_size 1 \
|
| 54 |
+
--results_file res9B \
|
| 55 |
+
--model open_flamingo \
|
| 56 |
+
--out_base_path /PATH/TO/Robust_mmfm/Results/open_flamingo \
|
| 57 |
+
--vision_encoder_path ViT-L-14 \
|
| 58 |
+
--checkpoint_path /PATH/TO/HUGGINGFACE/hub/models--openflamingo--OpenFlamingo-9B-vitl-mpt7b/snapshots/7e36809c73d038829ad5fba9d0cc949b4e180562/checkpoint.pt \
|
| 59 |
+
--lm_path anas-awadalla/mpt-7b \
|
| 60 |
+
--lm_tokenizer_path anas-awadalla/mpt-7b \
|
| 61 |
+
--precision float16 \
|
| 62 |
+
--cross_attn_every_n_layers 4 \
|
| 63 |
+
--coco_train_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/train2014 \
|
| 64 |
+
--coco_val_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/val2014 \
|
| 65 |
+
--coco_karpathy_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/karpathy_coco.json \
|
| 66 |
+
--coco_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/captions_val2014.json \
|
| 67 |
+
--coco_cf_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO_CF \
|
| 68 |
+
--flickr_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/Flickr30k/Images \
|
| 69 |
+
--flickr_karpathy_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/Flickr30k/karpathy_flickr30k.json \
|
| 70 |
+
--flickr_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/Flickr30k/dataset_flickr30k_coco_style.json \
|
| 71 |
+
--vizwiz_train_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/train \
|
| 72 |
+
--vizwiz_test_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/val \
|
| 73 |
+
--vizwiz_train_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/train_questions_vqa_format.json \
|
| 74 |
+
--vizwiz_train_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/train_annotations_vqa_format.json \
|
| 75 |
+
--vizwiz_test_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/val_questions_vqa_format.json \
|
| 76 |
+
--vizwiz_test_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/VizWiz/val_annotations_vqa_format.json \
|
| 77 |
+
--vqav2_train_image_dir_path /home/htc/kchitranshi/SCRATCH/COCO/train2014 \
|
| 78 |
+
--vqav2_train_questions_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_OpenEnded_mscoco_train2014_questions.json \
|
| 79 |
+
--vqav2_train_annotations_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_mscoco_train2014_annotations.json \
|
| 80 |
+
--vqav2_test_image_dir_path /home/htc/kchitranshi/SCRATCH/COCO/val2014 \
|
| 81 |
+
--vqav2_test_questions_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_OpenEnded_mscoco_val2014_questions.json \
|
| 82 |
+
--vqav2_test_annotations_json_path /home/htc/kchitranshi/SCRATCH/vqav2/v2_mscoco_val2014_annotations.json \
|
| 83 |
+
--textvqa_image_dir_path /mnt/datasets/textvqa/train_images \
|
| 84 |
+
--textvqa_train_questions_json_path /home/htc/kchitranshi/SCRATCH/RobustVLM/textvqa/train_questions_vqa_format.json \
|
| 85 |
+
--textvqa_train_annotations_json_path /home/htc/kchitranshi/SCRATCH/RobustVLM/textvqa/train_annotations_vqa_format.json \
|
| 86 |
+
--textvqa_test_questions_json_path /home/htc/kchitranshi/SCRATCH/RobustVLM/textvqa/val_questions_vqa_format.json \
|
| 87 |
+
--textvqa_test_annotations_json_path /home/htc/kchitranshi/RobustVLM/textvqa/val_annotations_vqa_format.json \
|
| 88 |
+
--ok_vqa_train_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/train2014 \
|
| 89 |
+
--ok_vqa_train_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/OpenEnded_mscoco_train2014_questions.json \
|
| 90 |
+
--ok_vqa_train_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/mscoco_train2014_annotations.json \
|
| 91 |
+
--ok_vqa_test_image_dir_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/COCO/val2014 \
|
| 92 |
+
--ok_vqa_test_questions_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/OpenEnded_mscoco_val2014_questions.json \
|
| 93 |
+
--ok_vqa_test_annotations_json_path /PATH/TO/Robust_mmfm/open_flamingo_datasets/OKVQA/mscoco_val2014_annotations.json \
|
bash/train_clip.sh
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
python vlm_eval/clip_train.py \
|
| 3 |
+
--num_epochs 1 \
|
| 4 |
+
--data_seeds 115 \
|
| 5 |
+
--data_name MS_COCO \
|
| 6 |
+
--method NONE \
|
| 7 |
+
--batch_size 128 \
|
| 8 |
+
--learning_rate 5e-7 \
|
| 9 |
+
--save_model \
|
| 10 |
+
--save_model_path ./fine_tuned_clip_models/NONE/
|
| 11 |
+
|
bash/train_clip_slurm.sh
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
#SBATCH --job-name=Search
|
| 3 |
+
#SBATCH --chdir=/home/htc/kchitranshi/ # Navigate to the working directory where your script lies
|
| 4 |
+
#SBATCH --output=/home/htc/kchitranshi/SCRATCH/%j.log # Standard output and error log
|
| 5 |
+
#
|
| 6 |
+
#SBATCH --gres=gpu:1
|
| 7 |
+
#SBATCH --cpus-per-task=12
|
| 8 |
+
#SBATCH --mem=100G
|
| 9 |
+
#SBATCH --partition=gpu # Specify the desired partition, e.g. gpu or big
|
| 10 |
+
#SBATCH --exclude=htc-gpu[020-023,037,038] # Only A40 GPU
|
| 11 |
+
#SBATCH --time=0-20:00:00 # Specify a Time limit in the format days-hrs:min:sec. Use sinfo to see node time limits
|
| 12 |
+
#SBATCH --ntasks=1
|
| 13 |
+
#
|
| 14 |
+
#SBATCH --mail-type=BEGIN
|
| 15 |
+
#SBATCH --mail-type=END
|
| 16 |
+
#SBATCH --mail-type=FAIL
|
| 17 |
+
#SBATCH --mail-user=
|
| 18 |
+
|
| 19 |
+
echo 'Getting node information'
|
| 20 |
+
date;hostname;id;pwd
|
| 21 |
+
|
| 22 |
+
echo 'Setting LANG to en_US.UTF-8'
|
| 23 |
+
LANG=en_US.UTF-8
|
| 24 |
+
|
| 25 |
+
which python
|
| 26 |
+
java -version
|
| 27 |
+
|
| 28 |
+
echo 'Enabling Internet Access'
|
| 29 |
+
export https_proxy=http://squid.zib.de:3128
|
| 30 |
+
export http_proxy=http://squid.zib.de:3128
|
| 31 |
+
|
| 32 |
+
echo 'Print GPUs'
|
| 33 |
+
/usr/bin/nvidia-smi
|
| 34 |
+
|
| 35 |
+
echo 'Running script'
|
| 36 |
+
cd Robust_mmfm
|
| 37 |
+
python vlm_eval/clip_train.py \
|
| 38 |
+
--num_epochs 1 \
|
| 39 |
+
--data_seeds 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 \
|
| 40 |
+
--data_name MS_COCO \
|
| 41 |
+
--method NONE \
|
| 42 |
+
--batch_size 128 \
|
| 43 |
+
--learning_rate 5e-7 \
|
| 44 |
+
--save_model \
|
| 45 |
+
--save_model_path ./fine_tuned_clip_models/NONE/
|