Я пытаюсь использовать torchviz, но он не дает никакого вывода. Нет ошибок или предупреждений ..
import numpy as np
import torch
from torchviz import make_dot
yhat= a+b*x_train_tensor
make_dot(yhat).render("attached", format="png")
Я также пытался использовать make_dot(yhat), но это тоже не работает. В приведенном выше примере создается только небольшой файл с именем «attach», но это не изображение. Очевидно, что x_train_tensor и yhat являются тензорами и прекрасно вычисляются. У меня проблемы с визуализацией.
Я уже видел эти сообщения.
Как мне визуализировать net в Pytorch?
https://github.com/szagoruyko/pytorchviz/issues/5
Но решить проблему не удалось. Я использую python 3.7.4 64 бит в vscode