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

..

__init__.py

Fix docs typo (#1355)

2020-04-03 07:35:09 +02:00

auto_mix_precision.py

improve partial Codecov (#1172)

2020-03-19 09:14:29 -04:00

callback_config.py

proper checkpoint implementation (#1043)

2020-03-04 23:02:19 -05:00

callback_hook.py

cleaning imports (#1032)

2020-03-12 12:41:37 -04:00

data_loading.py

generalize reinstantiation of dataloader (#1346)

2020-04-03 17:55:08 -04:00

deprecated_api.py

Simplify progress bar args (#1108)

2020-04-03 00:53:00 +02:00

distrib_data_parallel.py

Simplify progress bar args (#1108)

2020-04-03 00:53:00 +02:00

distrib_parts.py

Shubhamagarwal92 master (#1349)

2020-04-03 17:56:19 -04:00

evaluation_loop.py

Fix for incorrect run on the validation set with overwritten validation_epoch_end and test_end (#1353)

2020-04-03 09:25:32 -04:00

ignored_warnings.py

Dim 0 warning (#256)

2019-09-26 13:20:54 -04:00

logging.py

cleaning imports (#1032)

2020-03-12 12:41:37 -04:00

model_hooks.py

quick patch __code__ (#1352)

2020-04-03 08:40:02 -04:00

optimizers.py

Remove default optimizer, add None optimizer option (#1279)

2020-04-02 11:48:53 -04:00

supporters.py

fix incomplete RunningMean (#1309)

2020-03-30 18:28:31 -04:00

trainer.py

Shubhamagarwal92 master (#1349)

2020-04-03 17:56:19 -04:00

training_io.py

Checkpointing interval (#1272)

2020-03-30 18:37:02 -04:00

training_loop.py

Doc fixes (#1362)

2020-04-03 15:02:20 -04:00

training_tricks.py

nan detection and intervention (#1097)

2020-03-19 09:24:45 -04:00