RYDE-WORK/visual-med-alpaca
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/visual-med-alpaca.git synced 2026-02-08 17:56:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update Fine_tune_GIT_on_an_image_captioning_dataset.py

Browse Source
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
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
code/med-git/Fine_tune_GIT_on_an_image_captioning_dataset.py
View File

@ -19,7 +19,7 @@ validation_data_csv = ""
# path to the folder containing the training data images # path to the folder containing the training data images
validation_data_folder = "" validation_data_folder = ""
# save pretrained model to # save pretrained model to
output_dir = "" output_dir = "./med-git-base"
df = pd.read_csv(train_data_csv) df = pd.read_csv(train_data_csv)