Visualizing model training in Fastai/Pytorch

One of the challenges in model training is to be sure if our model is initialized correctly. We also want to debug/visualize the state of model. It helps in model training if we know what we are looking for. For e.g. if we know the model activations ...