Mdrnfox commited on
Commit
86b9502
·
verified ·
1 Parent(s): adc8c99

Update run_eval.py

Browse files
Files changed (1) hide show
  1. 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(device) // 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
  }
 
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
  }