Cannot log gpu memory without making on step = True while logging train loss

[karthi0804]

Discussed in #7582

^{Originally posted by karthi0804 May 18, 2021}
I am using 1.3.0 version. I wanted to just log the gpu memory at all time steps. So I enabled log_gpu_memory='all' and log_every_n_steps=1. I couldn't find the gpu logging for steps. But If I log the train loss on steps, I can see the gpu logging at each steps. Is this how it is intended to work? I expect that gpu logging should be independent of self.log().

[tchaton]

Dear @karthi0804,

Thanks for reporting this bug. Would you be interested to resolve this bug ? This should be quite easy.

The memory metrics are computed there and logged right after:
https://github.com/PyTorchLightning/pytorch-lightning/blob/b978d2a1f2f9cc9942971a6e252ccd1209a9269b/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py#L98

Best,
T.C

[karthi0804]

@tchaton thanks for fixing it! hopefully I will contribute next time.