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):
34
34
r"""Automatically monitors and logs device stats during training, validation and testing stage.
35
35
``DeviceStatsMonitor`` is a special callback as it requires a ``logger`` to passed as argument to the ``Trainer``.
36
36
37
-
38
37
**Logged Metrics**
39
38
40
39
Logs device statistics with keys prefixed as ``DeviceStatsMonitor.{hook_name}/{base_metric_name}``.
@@ -49,7 +48,7 @@ class DeviceStatsMonitor(Callback):
49
48
50
49
**CUDA GPU (via `torch.cuda.memory_stats`)**
51
50
52
- Logs memory statistics from PyTorch caching allocator (all in Bytes).
51
+ Logs memory statistics from PyTorch caching allocator (all in Bytes).
53
52
GPU compute utilization is not logged by default.
54
53
55
54
*General Memory Usage:*
You can’t perform that action at this time.
0 commit comments