Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/tests/models
History
Adrian Wälchli 02ac4b0b6a
Replace .get_model() with explicit .lightning_module (#6035) 
* rename get_model -> lightning_module

* update references to get_model

* pep8

* add proper deprecation

* remove outdated _get_reference_model

* fix cyclic import
 		2021-02-18 15:59:54 +01:00
..
conf
…
data [tests/models] refactor with BoringModel (#5507) 2021-02-11 14:32:07 +00:00
__init__.py
…
test_amp.py clean AMP logic (#5994) 2021-02-16 19:06:47 +00:00
test_cpu.py Remove torch<=1.4.0 checks (#5998) 2021-02-16 17:53:40 -05:00
test_gpu.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
test_grad_norm.py [tests/models] refactor with BoringModel (#5507) 2021-02-11 14:32:07 +00:00
test_hooks.py Add before_batch_transfer and after_batch_transfer hooks (#3671) 2021-02-18 06:58:12 -05:00
test_horovod.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
test_hparams.py [tests/models] refactor with BoringModel (#5507) 2021-02-11 14:32:07 +00:00
test_model_hooks.py Refactor simplify tests (#5861) 2021-02-08 11:52:02 +01:00
test_onnx.py [tests/models] refactor with BoringModel (#5507) 2021-02-11 14:32:07 +00:00
test_restore.py Replace .get_model() with explicit .lightning_module (#6035) 2021-02-18 15:59:54 +01:00
test_sync_batchnorm.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
test_torchscript.py Delete tests.helpers.TrialMNISTDataModule (#5999) 2021-02-18 03:35:38 +00:00
test_tpu.py [HotFix] Resolve TPU Training (#6027) 2021-02-17 16:40:13 +00:00