Skip to content
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

Open
Lijinxisa opened this issue Mar 8, 2023 · 1 comment
Open

PSA运行问题,急需大佬解答 #107

Lijinxisa opened this issue Mar 8, 2023 · 1 comment

Comments

@Lijinxisa
Copy link

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)

@SuperFDX
Copy link

channel=512就是输入通道,根据你的输入特征图改变。reduction=8应该是表示对输入特征张量的通道数压缩的程度,可以自己调看什么值对你好。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants