torchmeter-cli 0.1.0__tar.gz → 0.1.2__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/PKG-INFO +2 -2
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/README.md +1 -1
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/pyproject.toml +1 -1
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/pyproject.toml.orig +1 -1
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/src/torchmeter_cli/__init__.py +1 -1
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/src/torchmeter_cli/cli.py +9 -6
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/src/torchmeter_cli/__main__.py +0 -0
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/src/torchmeter_cli/benchmark.py +0 -0
- {torchmeter_cli-0.1.0 → torchmeter_cli-0.1.2}/src/torchmeter_cli/result.py +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.3
|
|
2
2
|
Name: torchmeter-cli
|
|
3
|
-
Version: 0.1.
|
|
3
|
+
Version: 0.1.2
|
|
4
4
|
Summary: Simple, reproducible performance benchmarking for PyTorch models.
|
|
5
5
|
Author: Pujan Neupane
|
|
6
6
|
Author-email: Pujan Neupane <pujanneupaneop0907@gmail.com>
|
|
@@ -18,7 +18,7 @@ TorchMeter makes it easy to measure the performance of PyTorch models and genera
|
|
|
18
18
|
|
|
19
19
|
## Status
|
|
20
20
|
|
|
21
|
-
🚧 **Early Development — v0.1.
|
|
21
|
+
🚧 **Early Development — v0.1.1**
|
|
22
22
|
|
|
23
23
|
TorchMeter is currently an experimental open-source project.
|
|
24
24
|
|
|
@@ -92,18 +92,21 @@ def _print_text(result) -> None:
|
|
|
92
92
|
print("─" * 32)
|
|
93
93
|
print(f"Model: {result.model_name}")
|
|
94
94
|
print(f"Device: {result.device}")
|
|
95
|
+
print(f"Warmup: {result.warmup_iterations}")
|
|
95
96
|
print(f"Iterations: {result.iterations}")
|
|
96
|
-
print()
|
|
97
|
+
print("─" * 32)
|
|
97
98
|
print("Latency")
|
|
98
|
-
print(f" Mean:
|
|
99
|
-
print(f" Median:
|
|
100
|
-
print(f" P95:
|
|
101
|
-
print(f" P99:
|
|
102
|
-
print()
|
|
99
|
+
print(f" Mean: {result.mean_latency_ms:.3f} ms")
|
|
100
|
+
print(f" Median: {result.median_latency_ms:.3f} ms")
|
|
101
|
+
print(f" P95: {result.p95_latency_ms:.3f} ms")
|
|
102
|
+
print(f" P99: {result.p99_latency_ms:.3f} ms")
|
|
103
|
+
print("─" * 32)
|
|
103
104
|
print(f"Throughput: {result.throughput_samples_per_second:.2f} samples/sec")
|
|
104
105
|
|
|
105
106
|
if result.peak_memory_mb is not None:
|
|
106
107
|
print(f"Peak memory: {result.peak_memory_mb:.2f} MB")
|
|
108
|
+
else:
|
|
109
|
+
print("Peak memory: N/A (CPU)")
|
|
107
110
|
|
|
108
111
|
print()
|
|
109
112
|
|
|
File without changes
|
|
File without changes
|
|
File without changes
|