Gpu mem (#308)

* Fixes #289 * Fixes #289 * added lbfgs support * Fixes #280 (#309) * added test seeds (#306) * added test seeds * added test seeds * updated docs * added lbfgs support (#310) * added lbfgs support * added lbfgs support * added lbfgs support * Fixes #280 (#309) * added test seeds (#306) * added test seeds * added test seeds * updated docs * added lbfgs support * added lbfgs support * added lbfgs support * added lbfgs support * added lbfgs support * added lbfgs support * added lbfgs support * added lbfgs support * Fixes #289 * Fixes #289 * merged master * merged master
2026-09-09 11:32:07 +08:00 · 2019-10-05 11:29:34 -04:00
parent 75fd89106f
commit 8f5a06bfb8
4 changed files with 46 additions and 9 deletions
@@ -127,7 +127,13 @@ trainer = Trainer(row_log_interval=10)
 Logs GPU memory when metrics are logged.   
 ``` {.python}
 # DEFAULT
-trainer = Trainer(log_gpu_memory=False)
+trainer = Trainer(log_gpu_memory=None)
+
+# log only the min/max utilization
+trainer = Trainer(log_gpu_memory='min_max')
+
+# log all the GPU memory (if on DDP, logs only that node)
+trainer = Trainer(log_gpu_memory='all')
 ```

 ---