-
Notifications
You must be signed in to change notification settings - Fork 22
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Multiscale Vision Transformers #40
Conversation
Codecov Report
@@ Coverage Diff @@
## main #40 +/- ##
============================================
- Coverage 100.00% 88.84% -11.16%
============================================
Files 49 52 +3
Lines 1091 1228 +137
============================================
Hits 1091 1091
- Misses 0 137 +137
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you please use Feedforward class from encoder/nn
as MLP Instead of 'common/mlp'?
Also you can import DropPath from timm library, use this import statement - from timm.models.layers import DropPath
,
Also if Pull Resquest is still in progress, you can change it to Draft.
Made the necessary changes |
|
||
def forward(self, x, thw_shape): | ||
x_block, thw_shape_new = self.attn(self.norm1(x), thw_shape) | ||
x_res, _ = attention_pool( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
here You need to import attention_pool from the attention module.
Co-authored-by: Abhijit Deo <[email protected]>
POOL_KV_STRIDE[i][0] | ||
] = KVQ_KERNEL |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
KVQ_KERNEL
this variable is not defined .
Co-authored-by: Abhijit Deo <[email protected]>
Co-authored-by: Abhijit Deo <[email protected]>
Co-authored-by: Abhijit Deo <[email protected]>
Co-authored-by: Abhijit Deo <[email protected]>
Co-authored-by: Abhijit Deo <[email protected]>
Co-authored-by: Abhijit Deo <[email protected]>
#14