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

Does this work suppor Pytorch model?

hheavenknowss opened this issue · comments

Hi there, this is a wonderful work, I can see model directly, I have not try but I wonder if this work support Pytorch model? if not, will it be supported in the future?

Right now it only supports tensorflow and only dense layers. I will look into more layer types and also pytorch and see how much effort it would be to implement. I will let you know.