Skip to content

AttributeError: 'torch._C.Node' object has no attribute 'ival' #80

Open
@bigcatMT

Description

@bigcatMT

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

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions