Hi there! Does anyone know a library to visualize pytorch’s tranformers?
thanks!
1 Like
Here’s a couple packages I’ve used to visualize transformers. Both of them end up plotting details that I personally didn’t require for my use case, so you might have to tweak your plots as well.
- PyTorchViz
- Netron (this requires converting your model to ONNX - this guide was really helpful)