lightning/tests/base
Nathan Raw 036bcea499
Call DataModule hooks implicitly in trainer ()
*  call dm hooks in trainer implicitly

*  update tests

* 📝 remove unused stage arg from dm docs

*  update tests

*  update tests

* 🚧 include stage in datamodule.setup

* 📝 docs

* 📝 docs

* added more dm tests

* added more dm tests

* 🐛 call dm.setup everywhere

* 🔥 pickle tests now implied by accelerator tests

* 🎨 set dm as attr of trainer

* 🐛 .

* 🚧 wip

* add can prepare test

* add can prepare test

* verified setup in fit

* fixed setup call

* fixed setup call

* fixed setup call

Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-08-01 20:17:57 -04:00
..
__init__.py Tests: refactor cleanup () 2020-05-10 13:15:28 -04:00
dataloaders.py Bugfix/_has_len () 2020-06-26 09:31:08 -04:00
datamodules.py Call DataModule hooks implicitly in trainer () 2020-08-01 20:17:57 -04:00
datasets.py fixing TPU tests () 2020-07-27 19:07:09 -04:00
deterministic_model.py EvalResult support for val loop (PR 3/5) () 2020-07-22 13:53:10 -04:00
develop_pipelines.py remove parameterize from TPU tests () 2020-07-09 06:46:07 -04:00
develop_utils.py fixing TPU tests () 2020-07-27 19:07:09 -04:00
mixins.py Fix lr key name in case of param groups () 2020-05-10 17:05:34 -04:00
model_optimizers.py replace Hparams by init args () 2020-05-24 18:59:08 -04:00
model_template.py Add onnx export () 2020-07-31 12:27:57 +02:00
model_test_dataloaders.py Add missing test for "multiple dataloader + percent_check fix" () 2020-06-23 11:21:24 -04:00
model_test_epoch_ends.py Add missing test for "multiple dataloader + percent_check fix" () 2020-06-23 11:21:24 -04:00
model_test_steps.py Tests: refactor cleanup () 2020-05-10 13:15:28 -04:00
model_train_dataloaders.py Bugfix/_has_len () 2020-06-19 23:38:15 -04:00
model_train_steps.py EvalResult support for val loop (PR 3/5) () 2020-07-22 13:53:10 -04:00
model_utilities.py repair CI for Win () 2020-06-26 21:38:25 -04:00
model_valid_dataloaders.py Add missing test for "multiple dataloader + percent_check fix" () 2020-06-23 11:21:24 -04:00
model_valid_epoch_ends.py EvalResult support for val loop (PR 3/5) () 2020-07-22 13:53:10 -04:00
model_valid_steps.py hotfix to unblock hparams and OmniConf - removes auto_register_init_args by default () 2020-05-31 08:29:51 -04:00
models.py Refactor model summary + generalize example input array () 2020-06-15 17:05:58 -04:00
test_datasets.py fixing TPU tests () 2020-07-27 19:07:09 -04:00