How can i modify the easy vqa model to train on DOC-vqa dataset ? here is the link to easy-vqa that was trained on simple shapes data. https://victorzhou.com/blog/easy-vqa/<