diff --git a/code/med-git/Fine_tune_GIT_on_an_image_captioning_dataset.py b/code/med-git/Fine_tune_GIT_on_an_image_captioning_dataset.py index e6997a2..d08725a 100644 --- a/code/med-git/Fine_tune_GIT_on_an_image_captioning_dataset.py +++ b/code/med-git/Fine_tune_GIT_on_an_image_captioning_dataset.py @@ -19,7 +19,7 @@ validation_data_csv = "" # path to the folder containing the training data images validation_data_folder = "" # save pretrained model to -output_dir = "" +output_dir = "./med-git-base" df = pd.read_csv(train_data_csv)