lightning/tests/tests_pytorch
Ray Schireman 0a5e75e8d1
Add `inference_mode` flag to Trainer (#15034)
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2022-10-12 12:22:01 +00:00
..
accelerators Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
benchmarks Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
callbacks Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
checkpointing Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
core Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
deprecated_api Error messages for the remaining callback hooks (#15064) 2022-10-11 19:18:47 +00:00
graveyard Error messages for removed DataModule hooks (#15072) 2022-10-11 21:58:05 +00:00
helpers Filter APEX future warning (#15078) 2022-10-11 13:04:48 +00:00
lite [Lite] precision_plugin -> precision (#15001) 2022-10-10 15:00:32 +00:00
loggers Move the `_scan_checkpoints` utility function (#9312) 2022-10-10 22:13:34 +00:00
loops Change `trainer.should_stop` to not stop in between an epoch and run until `min_steps/min_epochs` only (#13890) 2022-08-27 12:12:24 +00:00
models Mark AMP test as flaky (#15055) 2022-10-10 23:26:13 +02:00
overrides Standalone Lite: Remaining Utilities (#14492) 2022-09-07 15:25:23 +00:00
plugins Use `torch.testing.assert_close` everywhere (#15031) 2022-10-07 16:59:04 +02:00
profilers Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
serve Various test fixes (#15068) 2022-10-11 03:47:16 -04:00
strategies Add ColossalAI strategy (#14224) 2022-10-11 13:59:09 +02:00
trainer Add `inference_mode` flag to Trainer (#15034) 2022-10-12 12:22:01 +00:00
tuner Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
utilities Introduce base collective and main subclasses (#15016) 2022-10-07 19:53:19 +00:00
__init__.py Remove old test artifacts (#14574) 2022-09-07 10:09:59 -04:00
conftest.py Filter APEX future warning (#15078) 2022-10-11 13:04:48 +00:00
run_standalone_tasks.sh Prepare CI to run on 3090s (#14910) 2022-09-29 14:01:59 +00:00
run_standalone_tests.sh Prepare CI to run on 3090s (#14910) 2022-09-29 14:01:59 +00:00
test_cli.py Added support for custom parameters in subclasses of `SaveConfigCallback` (#14998) 2022-10-05 11:10:29 +02:00