Answers for "how can I plot model in pytorch"

1

how can I plot model in pytorch

from torchviz import make_dot

make_dot(yhat, params=dict(list(model.named_parameters()))).render("rnn_torchviz", format="png")
Posted by: Guest on March-25-2022

Python Answers by Framework

Browse Popular Code Answers by Language