Skip to content

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

Open
@Lijinxisa

Description

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)

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions