lightning/tests/tests_pytorch
Ishan Dutta 6cbc9dfb91
Remove NumPy from Callback scripts (#17267)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
2023-04-04 13:28:39 +00:00
..
accelerators Replace HPU with external implementation (#17067) 2023-03-24 13:30:53 +01:00
benchmarks Rename `replace_sampler_ddp|replace_sampler` to `use_distributed_sampler` (#16829) 2023-02-22 14:07:02 +01:00
callbacks Remove NumPy from Callback scripts (#17267) 2023-04-04 13:28:39 +00:00
checkpointing Mark the connectors as protected (#17008) 2023-03-10 00:44:04 +01:00
core Reroute LightningModule gradient clipping to Fabric (#16941) 2023-03-14 11:31:20 +00:00
deprecated_api copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
graveyard Replace HPU with external implementation (#17067) 2023-03-24 13:30:53 +01:00
helpers refactor: move RunIf to PL pkg (#16923) 2023-03-03 15:45:59 +00:00
loggers update for neptune 1.0 (#16761) 2023-03-21 19:00:07 +00:00
loops Error checking for non-iterables (#17007) 2023-03-10 17:02:28 +00:00
models Simplify the Trainer core logic (#17017) 2023-03-13 13:43:57 +00:00
overrides Consume the prediction batch indices iteratively (#16826) 2023-02-22 17:03:08 +01:00
plugins Replace HPU with external implementation (#17067) 2023-03-24 13:30:53 +01:00
profilers copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
serve tests: switch imports for pytorch (#16595) 2023-02-02 10:06:45 +00:00
strategies Generalize `Optimizer` validation to accommodate both FSDP 1.x and 2.x (#16733) 2023-03-28 12:32:21 +00:00
trainer Avoid `inference_mode` with `torch.compile` (#17215) 2023-03-29 19:43:28 +00:00
tuner Merge DDPStrategy and DDPSpawnStrategy in PL (#16809) 2023-02-27 14:43:23 -05:00
utilities Avoid `inference_mode` with `torch.compile` (#17215) 2023-03-29 19:43:28 +00:00
__init__.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
conftest.py Avoid `inference_mode` with `torch.compile` (#17215) 2023-03-29 19:43:28 +00:00
run_standalone_tasks.sh copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
run_standalone_tests.sh Always run standalone tests (#16705) 2023-02-20 14:58:44 +00:00
test_cli.py update for neptune 1.0 (#16761) 2023-03-21 19:00:07 +00:00