Shape visualization problem

waleedka / hiddenlayer

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

MIT License

1.79k stars 266 forks source link

Shape visualization problem #16

Closed wuyxiquanquan closed 5 years ago

wuyxiquanquan commented 5 years ago

Hi: Thank you for you amazing work! But I have a problem about the visualization:How can I get more shape info? Can you tell me the solution? best, yuxiang

vgg11 3

waleedka commented 5 years ago

Thanks for reporting this. I tracked the problem and just pushed a fix. Please get the latest from Github and that issue should be solved.