Skip to content

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

@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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions