Я пытаюсь сгенерировать визуальный график Re snet forward-pass. Я сделал следующее, чтобы увидеть график, но он показывает мне операции в обратном проходе. «rn50_attached.pdf» показывает все в Backward-pass с именами узлов, такими как Conv2dBackward, BatchNormBackward.
>>> rn50=torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet50_ssl')
>>> x = torch.zeros(1, 3, 224, 224, dtype=torch.float, requires_grad=False)
>>> make_dot(rn50(x)).render("rn50_attached")
'rn50_attached.pdf'
Сгенерированный PDF выглядит следующим образом.
Как визуализировать узлы в прямом проходе? Как их правильно визуализировать с помощью прямого именования?
Что такое Thnn в ThnnConv2dBackward?