You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
The text was updated successfully, but these errors were encountered:
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
The text was updated successfully, but these errors were encountered: