def __init__(self, img_dim, question_dim, **kwargs): super().__init__() self.image_attention_model = AttentionLayer(img_dim, question_dim, **kwargs) self.out_dim = self.image_attention_model.out_dim
def __init__(self, img_dim, question_dim, **kwargs): super(ImageFeatureEmbedding, self).__init__() self.image_attention_model = AttentionLayer(img_dim, question_dim, **kwargs) self.out_dim = self.image_attention_model.out_dim