lightning/tests/models
Adrian Wälchli ea5cfd2005
move batch to device before sending it to hooks (#7378)
* update train step

* test

* x

* limits

* val

* typeo

* x

* x

* step

* min gpus

* run all loops

* x

* limit test

* profiler

* clean up accelerator code

* move files

* rename

* move tests

* changelog

* reorder callbacks and model hooks

* add test description

* replace unneccessary method

* fix chlog

* adjust batch_to_device for DP Plugin

* update tests for dataloader idx

* unused imports

* hook change

* switch None

* clear memory

* change to None

* None

* None

* memory savings

* remove redundant todo

* hack

* cheat

* Revert "cheat"

This reverts commit a8433bd0b4.

* Revert "hack"

This reverts commit 43a6d1edeb.

* update new epoch loop

* remove from old loop code

* update chlog

* update hook test

* changelog

* teardown

* integrate changes in new eval loop

* fix hook calls

* add prediction step

* bad merge

* Revert "bad merge"

This reverts commit 488080863c.

* fix train batch hook test

* rm -rf _notebooks

* update chlog

* release memory

* fix type

* notebooks mess

* debug

* Revert "debug"

This reverts commit eec4ee2f77.

* teardown

* fix teardown bug

* debug

* x

* debug

* Revert "debug"

This reverts commit a6e6101946.

Revert "debug"

This reverts commit 5ddeaec069.

debug


debug


Revert "debug"

This reverts commit 605be746f7daedf265b2c05a1c153ce543394435.

Revert "Revert "debug""

This reverts commit a7612d5410409ed886cfb609457349ecf44cbfa8.

debug


x


x


x


s


tol


x


tol

* Fix changelog

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2021-07-05 09:31:39 +01:00
..
conf bugfix: Resolve interpolation bug with Hydra (#5406) 2021-01-26 14:29:47 +01:00
data deprecate hpc_load() and integrate it with restore() (#7955) 2021-06-14 12:20:01 +00:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_cpu.py Fix truncated_bptt_steps hiddens detach() and improve docs (#8145) 2021-07-01 22:16:14 +01:00
test_gpu.py Add torchelastic check when sanitizing GPUs (#8095) 2021-06-23 14:09:53 +02:00
test_grad_norm.py Enable logger connector re-design (#7891) 2021-06-09 14:24:45 +00:00
test_hooks.py move batch to device before sending it to hooks (#7378) 2021-07-05 09:31:39 +01:00
test_horovod.py Remove unnecessary use of comprehension (#8149) 2021-06-27 10:00:02 +01:00
test_hparams.py Support `save_hyperparameters()` in LightningModule dataclass (#7992) 2021-06-16 10:30:58 +02:00
test_onnx.py Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00:00
test_restore.py [bugfix] Clean Validation Sanity Checking metrics (#8171) 2021-06-28 13:49:56 -04:00
test_sync_batchnorm.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_torchscript.py [feat] Support custom filesystems in LightningModule.to_torchscript (#7617) 2021-05-21 11:23:15 +00:00
test_tpu.py Clean-up after logger connector redesign 2/2 (#7631) 2021-06-10 12:09:01 +00:00