wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-19 13:00:36 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
dac41030d48acbfecdf7c083b8e7b00f3fd9be06
pytorch-lightning/pytorch_lightning/models
T
History
Max Horn dac41030d4 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
2019-09-04 10:43:46 -04:00
..
__init__.py
fix rltv imports
2019-08-06 22:45:27 +02:00
trainer.py
Allow to deactivate GPU memory logging in Trainer (#190)
2019-09-04 10:43:46 -04:00