| python run_medclip.py \ | |
| --output_dir "model" \ | |
| --overwrite_output_dir \ | |
| --text_model_name_or_path="allenai/scibert_scivocab_uncased" \ | |
| --vision_model_name_or_path="openai/clip-vit-base-patch32" \ | |
| --tokenizer_name="allenai/scibert_scivocab_uncased" \ | |
| --mimic_data_dir="/home/shpotes/medclip/data/mimic-cxr/" \ | |
| --mimic_train_file="train_dataset.json" \ | |
| --mimic_validation_file="validate_dataset.json" \ | |
| --mimic_mode="docs" \ | |
| --roco_data_dir="/home/shpotes/medclip/data/roco-dataset/" \ | |
| --do_train --do_eval \ | |
| --num_train_epochs="18" \ | |
| --preprocessing_num_workers=32 \ | |
| --per_device_train_batch_size=64 \ | |
| --per_device_eval_batch_size=64 \ | |
| --warmup_steps=5000 \ | |
| --learning_rate="3e-4" |