How do I add histogram of weights and gradients of pre-trained ResNet in tensorboardX summary writer?