Hello,
Is there any interesting tool to plot the model architecture in 3D or 2D at least ?
Thank you
Hello,
Is there any interesting tool to plot the model architecture in 3D or 2D at least ?
Thank you
Tensorboard is one option, have a look at the tutorial for it:
https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html