[Feature] get maximum device memory allocated in train log

[FlyingQianMM]

In training log, maximum device memory allocated is shown.

[nepeplwu]

This PR conflicts with the latest code, please fix it

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}