waleedka / hiddenlayer

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

Fix deprecated torch method call to uniqueName #52

Closed jccurtis closed 4 years ago

jccurtis commented 5 years ago

Fixes #51

hiddenlayer==0.2.0 is not compatible with torch==1.2.0 (which is currently the latest installed with pip3 install torchvision and overrides an existing/lower torch version.) This PR is a simple patch which uses the unique() method of a torch._C.Value instead of the uniqueName method which is missing in torch==1.2.0.

waleedka commented 4 years ago

Thank you so much for the fix. And, apologies for the long delay. It's merged.