Skip to content

Commit df20e4b

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent fd323ed commit df20e4b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/lightning/pytorch/callbacks/device_stats_monitor.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff 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:*

0 commit comments

Comments
 (0)