I wonder how to visualize gradient by histogram with tensorboardX in pytorch
https://github.com/lanpa/tensorboardX/blob/9d2cbeb26778cc9784a6a7028e75b7fd0950ce87/demo.py#L44
Replace param with something like param.grad should be good to go.
param
param.grad
Most helpful comment
https://github.com/lanpa/tensorboardX/blob/9d2cbeb26778cc9784a6a7028e75b7fd0950ce87/demo.py#L44
Replace
paramwith something likeparam.gradshould be good to go.