Skip to content

Commit 4b43edf

Browse files
committed
Fix CI: Format scripts/measure_memory.py with ruff
1 parent d5fdb2b commit 4b43edf

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

scripts/measure_memory.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -93,9 +93,9 @@ def measure_comparison(
9393
"""
9494
Compare FP32 vs BitNet memory usage for a given configuration.
9595
"""
96-
print(f"\n{'='*80}")
96+
print(f"\n{'=' * 80}")
9797
print(f"Measuring: {model_name}, batch={batch_size}, image_size={image_size}×{image_size}")
98-
print(f"{'='*80}")
98+
print(f"{'=' * 80}")
9999

100100
# FP32 model
101101
print("Creating FP32 model...")
@@ -136,9 +136,9 @@ def print_summary_table(results: list[dict]) -> None:
136136
"""
137137
Print a summary table suitable for the paper.
138138
"""
139-
print(f"\n\n{'='*80}")
139+
print(f"\n\n{'=' * 80}")
140140
print("SUMMARY TABLE (for paper)")
141-
print(f"{'='*80}\n")
141+
print(f"{'=' * 80}\n")
142142

143143
print(f"{'Configuration':<40} {'FP32 Peak (MB)':>15} {'BitNet Peak (MB)':>16} {'Ratio':>8}")
144144
print("-" * 80)
@@ -220,9 +220,9 @@ def main() -> None:
220220
print_summary_table(results)
221221

222222
# Key insights
223-
print(f"\n\n{'='*80}")
223+
print(f"\n\n{'=' * 80}")
224224
print("KEY INSIGHTS")
225-
print(f"{'='*80}\n")
225+
print(f"{'=' * 80}\n")
226226
print("1. BitNet model parameters: ~16× smaller than FP32 (ternary weights)")
227227
print("2. BitNet training peak memory: Similar or HIGHER than FP32")
228228
print("3. Reason: FP32 gradients + STE overhead + activation memory")

0 commit comments

Comments
 (0)