Allow to deactivate GPU memory logging in Trainer (#190)

* Allow to deactivate GPU memory logging in Trainer

Adds the flag `log_gpu_memory` to Trainer to deactivate logging of GPU
memory utilization. On some servers logging the GPU memory usage can
significantly slow down training.

* Update Logging.md

* Update trainer.py
This commit is contained in:
Max Horn
2019-09-04 10:43:46 -04:00
committed by William Falcon
parent 0872c32151
commit dac41030d4
2 changed files with 14 additions and 1 deletions
+8
View File
@@ -14,6 +14,14 @@ Every k batches lightning will make an entry in the metrics log
``` {.python}
# DEFAULT (ie: save a .csv log file every 10 batches)
trainer = Trainer(add_log_row_interval=10)
```
---
#### Log metric row every k batches
Logs GPU memory when metrics are logged.
``` {.python}
# DEFAULT
trainer = Trainer(log_gpu_memory=False)
```
---