Net2Vis icon indicating copy to clipboard operation
Net2Vis copied to clipboard

Published 20 hours ago • verified publisher icon viscom-ulm

PyTorch implementation

Open Sparkier opened this issue 3 years ago • 1 comments

Describe the request It would be nice to also make Net2Vis available for pytorch users.

How can it be tackled? We can likely implement something similar to the keras translate with model.state_dict().

Sparkier avatar Apr 28 '21 07:04 Sparkier

I tried implementing something like this with this feature based on onnx conversion: https://github.com/viscom-ulm/Net2Vis/tree/feature/torch However, there are multiple problems with respect to performance and onnx conversion. I currently do not have the time to debug this.

Sparkier avatar Oct 21 '21 11:10 Sparkier