【发布时间】:2021-09-23 01:28:11
【问题描述】:
我想知道为什么 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,))
图片:
【问题讨论】:
-
不是 100% 确定,但它可能正在存储反向传播可能需要的最大索引
标签: tensorflow pytorch tensorboard resnet