search

Atcold / NYU-DLSP20

NYU Deep Learning Spring 2020
https://atcold.github.io/NYU-DLSP20/
Other
6.66k stars 2.22k forks source link

added keras-style model summaries for torch models and colormap to view MNIST images in greyscale #829

Closed RaoSukrit closed 1 year ago

RaoSukrit commented 1 year ago
  1. Added keras-style model summaries for torch models to view model graphs along with number of trainable params per layer (needs new package torchinfo
  2. Added color map when plotting images in matplotlib to view MNIST images in greyscale
review-notebook-app[bot] commented 1 year ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

Atcold commented 1 year ago

Thanks, but I'm okay with only the parameters count and the Viridis colourmap.