Fine-tuning pretrained VILBERT on a custom dataset
ir0nt0ad opened this issue · comments
ir0nt0ad commented
❓ Questions and Help
Hi, I want to use the Hateful Memes VILBERT checkpoint from the zoo to fine-tune it on my custom dataset. I run the following command, and it appears to work:
mmf_run config=mmf/projects/my_dataset/configs/vilbert/direct.yaml model=vilbert dataset=my_dataset checkpoint.resume_zoo=vilbert.finetuned.hateful_memes.direct checkpoint.resume_pretrained=True run_type=train_val
Just wanted to make sure that it actually does what I want it to do. Thanks!
ApoorvVerma5 commented
Hey, can you tell me how you converted your dataset "my_dataset" in the repo, share the script too if possible.