我正在使用火炬摘要
from torchsummary import summary
我想在打印模型摘要时传递多个参数，但这里提到的示例：pytorch中的模型总结只采取了一个论点。例如：
model = Network().to(device)
summary(model,(1,28,28))
原因是前向函数需要两个参数作为输入，例如：
def forward(self, img1, img2):
我如何在这里传递两个参数？
您可以使用此处给出的示例：pytorch总结多输入
summary(model, [(1, 16, 16), (1, 28, 28)])
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
