logo
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions

11 lines
282 B

"""
UNITER for VE model
"""
from .vqa import UniterForVisualQuestionAnswering
class UniterForVisualEntailment(UniterForVisualQuestionAnswering):
""" Finetune multi-modal BERT for VE
"""
def __init__(self, config, img_dim):
super().__init__(config, img_dim, 3)