External-Attention-pytorch icon indicating copy to clipboard operation
External-Attention-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon xmu-xiaoma666

关于输入输出参数的问题

Open KevinYang95 opened this issue 3 years ago • 0 comments

感谢您整理的内容，非常好，但是对于参数有一些问题，希望可以帮忙解答 from attention.DANet import DAModule import torch

input=torch.randn(50,512,7,7) →请问（512,7,7)这个是特征图尺寸对吧，50代表什么意思呢，batchsize？ danet=DAModule(d_model=512,kernel_size=3,H=7,W=7) → d_model代表什么呢 print(danet(input).shape)

KevinYang95 avatar Jun 29 '21 08:06 KevinYang95