Problem with output size using FoldDuplicates.

waleedka / hiddenlayer

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

MIT License

1.79k stars 266 forks source link

Problem with output size using FoldDuplicates. #22

Closed lorinczszabolcs closed 5 years ago

lorinczszabolcs commented 5 years ago

Hello @waleedka !

I have just noticed, that when I use FoldDuplicates, I get wrong output size (On the graph the output size after the first module is shown, instead of the output size of the last module of the folded duplicates.)

waleedka commented 5 years ago

Good catch. I just pushed a fix.