File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
src/lightning/pytorch/callbacks Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -34,7 +34,6 @@ class DeviceStatsMonitor(Callback):
3434 r"""Automatically monitors and logs device stats during training, validation and testing stage.
3535 ``DeviceStatsMonitor`` is a special callback as it requires a ``logger`` to passed as argument to the ``Trainer``.
3636
37-
3837 **Logged Metrics**
3938
4039 Logs device statistics with keys prefixed as ``DeviceStatsMonitor.{hook_name}/{base_metric_name}``.
@@ -49,7 +48,7 @@ class DeviceStatsMonitor(Callback):
4948
5049 **CUDA GPU (via `torch.cuda.memory_stats`)**
5150
52- Logs memory statistics from PyTorch caching allocator (all in Bytes).
51+ Logs memory statistics from PyTorch caching allocator (all in Bytes).
5352 GPU compute utilization is not logged by default.
5453
5554 *General Memory Usage:*
You can’t perform that action at this time.
0 commit comments