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

[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