我想知道为什么resnet50的MaxPool2d层在张力板上可视化时会输出2个张量?resnet50和代码的可视化如下图所示。谢谢!
代码:
import torchvision
import torch
from tensorboardX import SummaryWriter
import torch.nn as nn
model = torchvision.models.resnet50(pretrained=False)
dummy_input=torch.rand((10, 3, 224, 224))
with SummaryWriter(comment='Net1')as w:
w.add_graph(model, (dummy_input,))图片:

发布于 2021-07-14 13:03:20
ResNet中最大池旁边的块具有剩余连接:这意味着该块的输入在卷积层之间拆分,并将“直接”路径添加到该块的卷积层的输出中。
此剩余链接在此图中标记为"identity“:

https://stackoverflow.com/questions/68371781
复制相似问题