关于gnconv的用法
Mentholatum opened this issue · comments
作者您好!您提到可以直接将gnconv()
替换传统的3×3卷积,我想请教一个新手问题,传统的nn.Conv()
可以设置输入输出的特征通道数不同,我注意到gnconv()
中输入输出的通道数保持不变,如果想改变特征通道数,是否还需要额外的传统卷积操作?感谢您的解答!
你好,感谢对我们工作的关注。 我们的模块本身没有假设输入输出维数相同,只需要修改模块中的proj_out
的输出维度就能改变模型输出的维数。
谢谢您的解答!