julrog / nn_vis

A project for processing neural networks and rendering to gain insights on the architecture and parameters of a model through a decluttered representation.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add check for supported neural networks

julrog opened this issue · comments

Not all neural networks are supported. Add checks for valid layer types and other attributes of a neural network to check its validity.

Done
Supported layers are: Dense
Flatten are ignored.