If you've worked with PyTorch, you've probably used visualkeras or torchview to visualise your model architecture. Well, the same is now available for .NET. Introducing... TorchSharpVisual.
With TorchSharpVisual, you can turn your torch.nn.Module objects into high-quality PNGs or raw Graphviz files! I explain everything on these platforms:
Github: https://github.com/JacobGoodchild/TorchSharpVisual
NuGet: nuget.org/packages/TorchSharpVisual
Feel free to try it out! Thanks.
Top comments (0)