Spaces:
Sleeping
Sleeping
Update run_eval.py
Browse files- run_eval.py +1 -1
run_eval.py
CHANGED
@@ -147,7 +147,7 @@ for cfg in CONFIGS:
|
|
147 |
"model_id": adapter_repo,
|
148 |
"adapter_type": adapter_type,
|
149 |
"trainable_params": cfg.get("trainable_params"),
|
150 |
-
"peak_gpu_mem_mb": torch.cuda.max_memory_allocated(
|
151 |
"run_date": datetime.datetime.utcnow().isoformat(timespec="seconds"),
|
152 |
"commit_sha": subprocess.check_output(["git", "rev-parse", "HEAD"]).strip().decode(),
|
153 |
}
|
|
|
147 |
"model_id": adapter_repo,
|
148 |
"adapter_type": adapter_type,
|
149 |
"trainable_params": cfg.get("trainable_params"),
|
150 |
+
"peak_gpu_mem_mb": torch.cuda.max_memory_allocated() // 1024**2 if torch.cuda.is_available() else None,
|
151 |
"run_date": datetime.datetime.utcnow().isoformat(timespec="seconds"),
|
152 |
"commit_sha": subprocess.check_output(["git", "rev-parse", "HEAD"]).strip().decode(),
|
153 |
}
|