ZHANG-SHI-CHANG / PoolAttentionDenseNet

use PoolAttention in DenseNet Transition by Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About you test

rwqzcq opened this issue · comments

when I run PADenseNet.py, but get this error:
RuntimeError: Given groups=1, weight of size [64, 3, 3, 3], expected input[1, 64, 112, 112] to have 3 channels, but got 64 channels instead

Could you please tell me how to fix this problem?