개인용

torchviz

JunHan 2024. 8. 1. 20:27

https://discuss.huggingface.co/t/how-to-plot-models-using-torchviz-or-hiddenlayer/24392

 

How to plot models using torchviz or hiddenlayer

I am trying to plot models using torchviz and hiddenlayer but both gets errors. torchviz - GitHub - waleedka/hiddenlayer: Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras. hiddenlayer - GitHub - szagoruyko/pytorchviz: A small p

discuss.huggingface.co