Closed alexlenail closed 4 years ago
It fixes conv dims, but still has issues in tensor dims. https://github.com/alexlenail/NN-SVG/blob/f99ca4363f9b71398ce7504d72314244dddde43e/AlexNet.js#L186 https://github.com/alexlenail/NN-SVG/blob/f99ca4363f9b71398ce7504d72314244dddde43e/AlexNet.js#L190
This seems to work, in my quick anecdotal testing.