Right column name

src/app.py

@@ -17,13 +17,13 @@ def get_results(model_name: str, library: str, options: list, access_token: str)
     stages = {"model": [], "gradients": [], "optimizer": [], "step": []}
     for i, option in enumerate(data):
         for stage in stages:
-            stages[stage].append(option["Training using Adam"][stage])
-        value = max(data[i]["Training using Adam"].values())
+            stages[stage].append(option["Training using Adam (Peek vRAM)"][stage])
+        value = max(data[i]["Training using Adam (Peek vRAM)"].values())
         if value == -1:
             value = "N/A"
         else:
             value = convert_bytes(value)
-        data[i]["Training using Adam"] = value
+        data[i]["Training using Adam (Peek vRAM)"] = value
 
     if any(value != -1 for value in stages["model"]):
         out_explain = "## Training using Adam explained:\n"

src/model_utils.py

@@ -97,7 +97,7 @@ def calculate_memory(model: torch.nn.Module, options: list):
                 "dtype": dtype,
                 "Largest Layer or Residual Group": dtype_largest_layer,
                 "Total Size": dtype_total_size,
-                "Training using Adam": dtype_training_size,
+                "Training using Adam (Peek vRAM)": dtype_training_size,
             }
         )
     return data