waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
MIT License
1.79k stars 266 forks source link

Fix TypeError for Pytorch Model #94 #97

Closed julien-blanchon closed 8 months ago

julien-blanchon commented 2 years ago

Catch TypeError exception raise by torch.onnx._optimize_trace with Pytorch 1.13.0. Fix https://github.com/waleedka/hiddenlayer/issues/94

julien-blanchon commented 2 years ago

At least it work (but graph without opt are pretty ugly 😢)

julien-blanchon commented 8 months ago

Close as inactive