HarisIqbal88 / PlotNeuralNet

Latex code for making neural networks diagrams
MIT License
21.93k stars 2.86k forks source link

Major bug: to_ConvConvRelu prints just two values from n_filer and width instead of all of them. #81

Open chrismaliszewski opened 4 years ago

chrismaliszewski commented 4 years ago

Hi.

I've just downloaded the latest version of the code to implement VGG16 graph on my own. I happened to encounter a bug when trying to display 3 Conv layers one after another using to_ConvConvRelu as, in result, only two layers are being displayed instead.

tikzeng.py file in, at least, two lines is wrong which causes the problem; these are lines 67 & 72.

Here is the fix to the problem: 67: xlabel={{ """+ (','.join(['"'+str(filer)+'"' for filer in n_filer])) + """ }}, 72: width={ """+ (','.join([str(w) for w in width])) + """ },

I just fixed this problem but it may happen in other places in the code.

Cheers.