File size: 747 Bytes
b45f0b4 237918a 98c1ffc b45f0b4 98c1ffc b45f0b4 237918a 98c1ffc 237918a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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" |