search

Yushi-Hu / PromptCap

natual language guided image captioning
73 stars 7 forks source link

training code #1

Closed Sino0904 closed 9 months ago

Sino0904 commented 1 year ago

Will the code for training the model be provided? thanks!

Yushi-Hu commented 1 year ago

Sure, we will release the training data soon! The training pipeline is in OFA, based on Fairseq

Yushi-Hu commented 9 months ago

The training codes are in this repo: https://github.com/OFA-Sys/OFASys. Please follow their fine-tuning instructions. We find eventually it is really hard to incorporate this whole repo into ours so we give up on that plan.