From 7438487e17948600ea3db9b3aa85a279b8772126 Mon Sep 17 00:00:00 2001 From: "Baian Chen (Andrew)" Date: Fri, 14 Apr 2023 17:31:55 +0800 Subject: [PATCH] Update Fine_tune_GIT_on_an_image_captioning_dataset.py --- code/med-git/Fine_tune_GIT_on_an_image_captioning_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)