Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

PyTorch implementation #10

Open
Sparkier opened this issue Apr 28, 2021 · 1 comment
Open

PyTorch implementation #10

Sparkier opened this issue Apr 28, 2021 · 1 comment
Labels
feature-request

Comments

@Sparkier
Copy link

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 Sparkier self-assigned this Apr 28, 2021
@Sparkier Sparkier mentioned this issue Oct 21, 2021
Closed
@Sparkier
Copy link
Author

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 Sparkier removed their assignment Oct 25, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
feature-request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@Sparkier