Which logging option to use?

6

NICE!

BTW MLFlow still has some limitations: Using log_gpu_memory with MLFLow logger causes an exception. · Issue #4411 · Lightning-AI/lightning · GitHub

I hope they resolve it soon.