We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b1a9c4f commit 0bcef4fCopy full SHA for 0bcef4f
src/llmcompressor/utils/metric_logging.py
@@ -17,12 +17,7 @@
17
18
class CompressionLogger:
19
"""
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
+ Log metrics related to compression algorithms
26
27
28
def __init__(self, module: torch.nn.Module):
0 commit comments