Understanding Fast.ai magic: added layers

Hi,

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

1 Like