Spaces:

nanotron
/

predict_memory

Running

nouamanetazi HF Staff commited on Feb 7, 2025

Commit

ed9dd0d

1 Parent(s): 4e75254

update

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -152,7 +152,6 @@ def calculate_memory_components(
             "ZeRO-3 Buffers": zero3_buffers,
             "Overhead": overhead,
             "Activations": activs,
-            "num_params_in_B": num_params_in_B
         },
         "Aggregates": {
             "Memory Before Optimizer States": memory_usage_before_optimstates,

             "ZeRO-3 Buffers": zero3_buffers,
             "Overhead": overhead,
             "Activations": activs,
         },
         "Aggregates": {
             "Memory Before Optimizer States": memory_usage_before_optimstates,