How can I retrain VisualBERT on another dataset?

  Kiến thức lập trình

I am looking to retrain the Vision-Language model VisualBERT on a specific dataset (images + text). How can I do this?

Out-of-the-box VisualBERT is trained on the COCO dataset, and I’d like to retrain it on my own data so that it retains the original learned model parameters, whilst being updated with the new data.

I’m simply unsure where the section of code is I need to change etc.

VisualBERT GitHub repo: https://github.com/uclanlp/visualbert

Thanks in advance.

LEAVE A COMMENT