lightning/tests/models
Adrian Wälchli a6aa1a0f82
make gpus=str in Trainer consistent with command line parsing of string (#6388)
* string gpu input

* update docs

* deprecation warning

* Revert "update docs"

This reverts commit c5f3893413.

* deprecation

* add changelog

* update parser

* update warning

* implement v1.5 behavior ahead of time

* formatting

* set accelerator in test to avoid different warning

* add warning

* remove todo warn

* Update pytorch_lightning/utilities/device_parser.py

Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>

* resolve flake8

Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: tchaton <thomas@grid.ai>
2021-05-04 09:56:27 +00:00
..
conf
data [bug] Update broadcast + reduce decision ModelCheckpoint] (#6410) 2021-03-14 17:14:27 +00:00
__init__.py
test_amp.py Remove `trainer.fit` return value [2/n] (#7237) 2021-04-28 19:11:32 +01:00
test_cpu.py Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00:00
test_gpu.py make gpus=str in Trainer consistent with command line parsing of string (#6388) 2021-05-04 09:56:27 +00:00
test_grad_norm.py
test_hooks.py Fix inconsistent outputs in `on_*_end` and `*_end` (#6969) 2021-04-13 15:16:21 +01:00
test_horovod.py Remove `trainer.fit` return value [2/n] (#7237) 2021-04-28 19:11:32 +01:00
test_hparams.py fix save_hyperparameters(container) if container is empty (#7268) 2021-04-30 13:38:42 +01:00
test_onnx.py Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00:00
test_restore.py update test for resume_from_checkpoint on missing file (#7255) 2021-05-04 09:16:34 +00:00
test_sync_batchnorm.py introduce default cluster environment for lightning-specific ddp (#5915) 2021-03-05 01:47:29 +00:00
test_torchscript.py Refactor: Runif for TPU and Horovod 5/n (#6301) 2021-03-02 16:21:20 +00:00
test_tpu.py Device updates for TPU Pod (#7243) 2021-04-30 23:14:06 +05:30