github-starred/TinyTorch
2
0
Fork 0
mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-04-29 19:52:36 -05:00
Code Issues 7 Packages Projects Releases Wiki Activity

Clean up formatting in verification functions

Browse Source
This commit is contained in:
Vijay Janapa Reddi
2025-12-05 12:12:38 -08:00
parent a5bfffe48f
commit 3aa6a9b040
3 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/16_compression/compression.py
View File

@@ -1381,7 +1381,7 @@ def verify_pruning_works(model, target_sparsity=0.8):
assert verified, f"Sparsity target not met: {sparsity:.2f} vs {target_sparsity:.2f}"
print(f"\n✅ VERIFIED: {sparsity*100:.1f}% sparsity achieved")
print(f"⚠️ Memory saved: 0 MB (dense numpy arrays)")
print(f"⚠️ Memory saved: 0 MB (dense numpy arrays)")
print(f"💡 LEARNING: Compute savings ~{sparsity*100:.1f}% (skip zero multiplications)")
print(f" In production: Use sparse formats (scipy.sparse.csr_matrix) for memory savings")