lightning/tests/models
Carlos Mocholí 5f920dc088
Refactor Horovod NCCL check (#11948)
2022-02-28 10:45:32 +00:00
..
conf bugfix: Resolve interpolation bug with Hydra (#5406) 2021-01-26 14:29:47 +01:00
data Mark `CheckpointConnector` as protected (#11550) 2022-02-03 02:26:08 +00:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py Use a dataclass as the scheduler config (#11443) 2022-01-18 20:23:32 +01:00
test_cpu.py Mark `CheckpointConnector` as protected (#11550) 2022-02-03 02:26:08 +00:00
test_gpu.py Rewrite accelerator_connector (#11448) 2022-02-17 23:38:39 +00:00
test_grad_norm.py Refactor `grad_norm` function (#9742) 2021-09-30 02:54:08 +00:00
test_hooks.py Avoid loading dataloaders if `limit_batches=0` (#11576) 2022-02-22 11:33:53 +00:00
test_horovod.py Refactor Horovod NCCL check (#11948) 2022-02-28 10:45:32 +00:00
test_hparams.py Fix `save_hyperparameters` when no parameters need saving (#11827) 2022-02-09 23:10:14 +00:00
test_onnx.py Avoid the deprecated `onnx.export(example_outputs=...)` in torch 1.10 (#11116) 2021-12-17 10:11:11 +01:00
test_restore.py Deprecate callback hooks `on_pretrain_routine_{start,end}` (#11794) 2022-02-24 19:32:48 +00:00
test_torchscript.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
test_tpu.py Remove `Trainer._device_type` (#11992) 2022-02-22 13:02:13 +00:00