Skip to content
This repository was archived by the owner on Jul 18, 2024. It is now read-only.

Conversation

@Drajan
Copy link

@Drajan Drajan commented May 6, 2019

Add an "output_mode" to VQA_Attention to enable it to switch between returning only the attention-weighted Image or return a concatenated vector of (attn Image + Question).

if(self.output_mode == 'Image'):
# Output attention-weighted image encodings
outputs = attention_enc_img
elif(self.output_mode == 'None'):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think 'None' is misleading - seems that the component will return something, right...?

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, will change this :) thanks!

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hey! I fixed this 'output_mode' to take two states: (1) 'Image' --- to return only attention-weighted image, and (2) 'Fusion' -- to return a concatenated output of attention-weighted image and question.

Hope this can be merged now :)

@tkornuta-ibm tkornuta-ibm merged commit 1b9929f into IBM:develop May 6, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants