小编you*_*lee的帖子

我尝试将 Pytorch 中的 DenseNet 逐层分析为 caffe-time 工具。

第一次试用：使用 autograd.profiler 如下所示

...
model = models.__dict__['densenet121'](pretrained=True)
model.to(device)

with torch.autograd.profiler.profile(use_cuda=True) as prof:
    model.eval()
print(prof)
...

但除了此消息外，任何结果都会显示：

<unfinished torch.autograd.profile>

最终，我想分析网络架构（igDenseNet）以检查瓶颈发生的位置。

任何人都可以这样做吗？