Change _optimize_trace to _optimize_graph

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

Open amitbar05 opened 1 year ago

amitbar05 commented 1 year ago

Fix #100

anwai98 commented 7 months ago

Hi @waleedka @philferriere,

Do you think this can be merged and make a release with this? (quite a few popular workflows make use of this feature)

(Thanks for this awesome work btw!)