lightning/pytorch_lightning/callbacks
Rohit Gupta 24809b0b26
Refactor GPUStatsMonitor to improve training speed (#3257)
* Refactor GPUMonitor to improve training speed

* added gpu ids to monitor

* update tests

* added deprecation warning

* pep

* fix test

* fix docs

* fix log_gpu_memory

* move deprecation check

* chlog

* Update CHANGELOG.md

* suggestions and fix

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2020-09-04 06:02:16 -04:00
..
__init__.py Refactor GPUStatsMonitor to improve training speed (#3257) 2020-09-04 06:02:16 -04:00
base.py callback method for on_save_checkpoint (#2501) 2020-08-28 16:50:52 +02:00
early_stopping.py bugfix/3185 transpose (#3252) 2020-09-01 09:17:52 -04:00
gpu_stats_monitor.py Refactor GPUStatsMonitor to improve training speed (#3257) 2020-09-04 06:02:16 -04:00
gpu_usage_logger.py Refactor GPUStatsMonitor to improve training speed (#3257) 2020-09-04 06:02:16 -04:00
gradient_accumulation_scheduler.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
lr_logger.py Changed LearningRateLogger to LearningRateMonitor (#3251) 2020-09-03 18:17:15 +00:00
lr_monitor.py Changed LearningRateLogger to LearningRateMonitor (#3251) 2020-09-03 18:17:15 +00:00
model_checkpoint.py use fsspec instead of gfile for all IO (#3320) 2020-09-03 14:19:20 +02:00
progress.py ref: restore on_eval_start hook (#3183) 2020-08-26 00:45:43 -04:00