Files
pytorch-lightning/pytorch_lightning
William FalconandShubham Agarwal 16f4cc9ff0 Shubhamagarwal92 master (#1349)
* SA: for #958: set torch cuda device when finding root

* SA: for #958: removing root gpu hack in trainer/evaluation_loop

* SA: setting torch cuda device

* comment line too long

* check if root gpu exists or available

* Incorporating suggestions on #1094

* since root gpu returns none instead of -1 for cpu

* undo changes

* fixed dp memory thing

Co-authored-by: Shubham Agarwal <shubhamagarwal92@gmail.com>
2020-04-03 17:56:19 -04:00
..
2020-03-30 18:37:02 -04:00
2020-04-03 15:03:00 -04:00
2020-03-31 08:57:48 -04:00
2020-04-03 17:56:19 -04:00
2020-03-31 08:57:48 -04:00