search

waleedka / hiddenlayer

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

Bugfix/get trace graph #67

Closed nanohanno closed 4 years ago

nanohanno commented 4 years ago

Builds on the earlier fix by jccurtis. This pull request fixes issue #66 . It makes hiddenlayer work with Pytorch 1.4, where get_trace_graph became private and ONNX formatting appears to have changed.

iNLyze commented 4 years ago

Works for me with PyTorch 1.4.0

RossAidan commented 4 years ago

Can this be merged in

waleedka commented 4 years ago

Thank you for the fix, and sorry for the delay. This is now merged.

twangnh commented 4 years ago

hi, but in pytorch 1.5, the problem of no attribute get trace graph happens again