Beispiel #1
0
    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
Beispiel #2
0
    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