Skip to content

Commit 0bcef4f

Browse files
committed
docstring
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent b1a9c4f commit 0bcef4f

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

src/llmcompressor/utils/metric_logging.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,7 @@
1717

1818
class CompressionLogger:
1919
"""
20-
Log metrics related to compression algorithm
21-
22-
:param start_tick: time when algorithm started
23-
:param losses: loss as result of algorithm
24-
:param gpu_type: device manufacturer (e.g. Nvidia, AMD)
25-
:param visible_ids: list of device ids visible to current process
20+
Log metrics related to compression algorithms
2621
"""
2722

2823
def __init__(self, module: torch.nn.Module):

0 commit comments

Comments
 (0)