lightning/tests/models
Krishna Kalyan 6586dd23b7
Mark `CheckpointConnector` as protected (#11550)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-02-03 02:26:08 +00:00
..
conf
data Mark `CheckpointConnector` as protected (#11550) 2022-02-03 02:26:08 +00:00
__init__.py
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 Deprecate Trainer.training_type_plugin in favor of trainer.strategy (#11141) 2021-12-22 02:11:43 +00:00
test_grad_norm.py Refactor `grad_norm` function (#9742) 2021-09-30 02:54:08 +00:00
test_hooks.py Add `LightningModule.lr_scheduler_step` (#10249) 2022-01-12 03:53:49 +00:00
test_horovod.py Deprecate Trainer.training_type_plugin in favor of trainer.strategy (#11141) 2021-12-22 02:11:43 +00:00
test_hparams.py Update pre-commit hook versions (#11202) 2021-12-22 17:09:27 +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 Mark `CheckpointConnector` as protected (#11550) 2022-02-03 02:26:08 +00:00
test_sync_batchnorm.py Update strategy import statements (#11231) 2021-12-23 08:26:28 +01:00
test_torchscript.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
test_tpu.py Update strategy import statements (#11231) 2021-12-23 08:26:28 +01:00