How to retrain VisualBERT on another dataset?

[Tess314]

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

I'm unsure what section of the code is I need to change.

Jupyter notebook I'm using: https://github.com/huggingface/transformers/blob/main/examples/research_projects/visual_bert/demo.ipynb

Thanks in advance.