-
Notifications
You must be signed in to change notification settings - Fork 1.9k
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
PSA运行问题,急需大佬解答 #107
Comments
channel=512就是输入通道,根据你的输入特征图改变。reduction=8应该是表示对输入特征张量的通道数压缩的程度,可以自己调看什么值对你好。 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
PSA下面这段代码里面的channel和reduction分别是什么意思,本人是新手,望大佬指导一下,不知道如何设定成自己想要的
from model.attention.PSA import PSA
import torch
input=torch.randn(50,512,7,7)
psa = PSA(channel=512,reduction=8)
output=psa(input)
print(output.shape)
The text was updated successfully, but these errors were encountered: