You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hello, I loaded pretrained model by the following provided code.
from mmf.common.registry import registry
model_cls = registry.get_model_class("vilbert")
model = model_cls.from_pretrained("vilbert.finetuned.hateful_memes.direct")
Now, I want to use this model on my dataset but I can't figure out what are the inputs to be provided in the model. These all are the inputs to be given,
input_0, input_1, targets, input_mask, segment_ids, return_output_dict, image_info_0 , image_features_0
The text was updated successfully, but these errors were encountered:
Hello, I loaded pretrained model by the following provided code.
from mmf.common.registry import registry
model_cls = registry.get_model_class("vilbert")
model = model_cls.from_pretrained("vilbert.finetuned.hateful_memes.direct")
Now, I want to use this model on my dataset but I can't figure out what are the inputs to be provided in the model. These all are the inputs to be given,
input_0, input_1, targets, input_mask, segment_ids, return_output_dict, image_info_0 , image_features_0
The text was updated successfully, but these errors were encountered: