Hi,
I think this (Visualizing your network in PyTorch) or this (GitHub - lanpa/tensorboardX: tensorboard for pytorch (and chainer, mxnet, numpy, ...)) will work.
Hi,
I think this (Visualizing your network in PyTorch) or this (GitHub - lanpa/tensorboardX: tensorboard for pytorch (and chainer, mxnet, numpy, ...)) will work.