Add a visualization module during Training

[SunshineSki]

Excellent Work！
It would be more helpful to add a visualization module for parameters during training.

[fracape]

Hi. The main purpose of this library/platform is to provide model implementation and evaluation pipelines. The provided training example serves as a "template". Each researcher has preferences in training monitoring tools, so please add your favorite dependencies to monitor your training.