File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed
Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments