GitLab

* add learning rate visualization

* support momentum visualization

* fix second axis font size error

* fix argparse description
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* fix lint

* Accelerate visualization

* Support custom batchsize and log-interval

Support custom batchsize and log-interval

* remove unnecessary keys and rename some varaible
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>