Open
Description
from torchvision.models.resnet import resnet50
import tensorwatch as tw
model = resnet50()
tw.draw_model(model, [1,3,512,512])
when using tensorwatch and jupyter to watch pytorch models as above codes show, report error as below:
module 'torch.onnx' has no attribute 'set_training'
then modify 'set_training' in /anaconda3/lib/python3.9/site-packages/tensorwatch/model_graph/hiddenlayer/summary_graph.py to
'select_model_mode_for_export', but report another error as below:
'torch._C.Node' object has no attribute 'ival'
related versions:
pytorch 1.10.1
tensorwatch 0.9.1
Metadata
Metadata
Assignees
Labels
No labels