lightning/tests/base
Nima Sarang 793036d29c
Support returning python scalars in DP (#1935)
* Override the default gather method to support scalars

* add computing average of a list

* bug: change if to elif

* add some tests

* change style

* change documentation

* use apply_to_collection in DP gather

* use apply_to_collection in DP gather

* fix warning msg

* override gather method in DP

* add tests for python scalars

* add python scalars to docstring

* Update message

* override gather method in DP

* formatting

* chlog

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai>
2020-08-07 09:18:29 +02:00
..
__init__.py
dataloaders.py Bugfix/_has_len (#2307) 2020-06-26 09:31:08 -04:00
datamodules.py updated sync bn (#2838) 2020-08-06 01:12:11 +02:00
datasets.py clean tests imports (#2834) 2020-08-06 16:58:51 +02:00
deterministic_model.py clean tests imports (#2834) 2020-08-06 16:58:51 +02:00
develop_pipelines.py remove parameterize from TPU tests (#2561) 2020-07-09 06:46:07 -04:00
develop_utils.py clean tests imports (#2834) 2020-08-06 16:58:51 +02:00
mixins.py
model_optimizers.py Bugfix: Lr finder and hparams compatibility (#2821) 2020-08-07 00:34:48 +02:00
model_template.py Add onnx export (#2596) 2020-07-31 12:27:57 +02:00
model_test_dataloaders.py Add missing test for "multiple dataloader + percent_check fix" (#2226) 2020-06-23 11:21:24 -04:00
model_test_epoch_ends.py Add missing test for "multiple dataloader + percent_check fix" (#2226) 2020-06-23 11:21:24 -04:00
model_test_steps.py
model_train_dataloaders.py Bugfix/_has_len (#2293) 2020-06-19 23:38:15 -04:00
model_train_steps.py Support returning python scalars in DP (#1935) 2020-08-07 09:18:29 +02:00
model_utilities.py repair CI for Win (#2358) 2020-06-26 21:38:25 -04:00
model_valid_dataloaders.py Add missing test for "multiple dataloader + percent_check fix" (#2226) 2020-06-23 11:21:24 -04:00
model_valid_epoch_ends.py Support returning python scalars in DP (#1935) 2020-08-07 09:18:29 +02:00
model_valid_steps.py
models.py Refactor model summary + generalize example input array (#1773) 2020-06-15 17:05:58 -04:00
test_datasets.py fixing TPU tests (#2632) 2020-07-27 19:07:09 -04:00