Update Fine_tune_GIT_on_an_image_captioning_dataset.py

This commit is contained in:
Baian Chen (Andrew) 2023-04-14 17:31:55 +08:00 committed by GitHub
parent e1064626cf
commit 7438487e17
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)