3 Commits

Author	SHA1	Message	Date
Carlos Mocholí	908382f196	Split GPUStatsMonitor function (#3644) ... * Split function * Add docstrings * Add typing annotations * Minor refactor * Make static to add a test	2020-09-25 07:30:30 +02:00
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
Rohit Gupta	f03943ee94	Fix GpuUsageLogger to work on different platforms (#3008) ... * Fix GpuUsageLogger * docstrings * misconfigexception * add basic tests * skip doctest * fix parameter and docstring * rm cl * skip doctest * cleanup * chlog * add suggestions from review * add test from suggestions * fix import * fix test * fix test * fix test * fix test * rename GpuUsageLogger to GPUStatsMonitor * doc fix * Apply suggestions from code review * update docs format * update docs * miss * merge * fix title formatting * unindent * punctuation * simplify if statements * fix test * suggestions * pep * Update CHANGELOG.md Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * fix on_train_batch_* * use AttributeDict * usage * rank zero Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * import * minor changes Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> Co-authored-by: Adrian Wälchli <adrian.waelchli@inf.unibe.ch> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>	2020-08-27 19:50:32 +02:00