wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-26 14:11:01 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
7ac1580a3153c6ab20b613101dfe88a95f1ad4a7
pytorch-lightning/pytorch_lightning
T
History
Allard Hendriksen 7ac1580a31 Add automatic GPU choice to trainer (#1426) 
* Add automatic GPU choice to trainer

This commit adds the `gpu_choice` parameter to Trainer. By default,
this parameter is set to 'manual' which causes no observable
difference in behavior.

When `gpu_choice` is set to "auto" and `gpus` is an int, then the
trainer will automatically allocate the first available GPU.
This is especially useful when GPUs are configured to be in "exclusive
mode", which means that only one process at a time can use them.

* Rename gpu_choice -> auto_select_gpus
2020-04-10 11:45:29 -04:00
..
callbacks
add rank warning (#1428)
2020-04-09 14:05:46 -04:00
core
add rank warning (#1428)
2020-04-09 14:05:46 -04:00
loggers
Fix TrainsLogger doctest failing (switch to bypass mode in GitHub CI) (#1379)
2020-04-08 11:52:52 -04:00
logging
add rank warning (#1428)
2020-04-09 14:05:46 -04:00
overrides
add rank warning (#1428)
2020-04-09 14:05:46 -04:00
profiler
Profiler summary (#1259)
2020-03-31 08:57:48 -04:00
pt_overrides
add rank warning (#1428)
2020-04-09 14:05:46 -04:00
root_module
add rank warning (#1428)
2020-04-09 14:05:46 -04:00
trainer
Add automatic GPU choice to trainer (#1426)
2020-04-10 11:45:29 -04:00
utilities
add rank warning (#1428)
2020-04-09 14:05:46 -04:00
__init__.py
Update __init__.py
2020-04-10 08:43:46 -04:00