Fixed tensor shapes for PyTorch models

waleedka / hiddenlayer

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

MIT License

1.79k stars 266 forks source link

Open ahrnbom opened 3 years ago

ahrnbom commented 3 years ago

Fixed shapes for PyTorch plots. See https://github.com/waleedka/hiddenlayer/issues/83 for more info.

githubrandomuser2017 commented 2 years ago

@waleedka Can you please merge this pull request?