Tensor width and height order seems to to be incorrect. (For filter size dimensions, it is not wrong)
<img width="1273" alt="Screen Shot 2023-01-14 at 23 24 59" src="https://user-images.githubusercontent.com/85290719/212495430-5d7e2f4b-3fa3-4922-a481-f32b3e0a6d29.png">
Hi @alexlenail. The text underneath "Architecture" is correct, as you can see that a height of 50 and a width of 500 gives rectangles that are wider than they are tall, but the labels are simply misplaced. The 50 label should be put on the height, not the width. Fixed in PR #60.