feat(pt): log pre-clip gradient total_norm and per-param norm to tensorboard#5252

iProzd merged 2 commits intodeepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

OutisLi:pr/tboardOutisLi/deepmd-kit:pr/tboardCopy head branch name to clipboard