├── .actions ├── assistant.py ├── pull_legacy_checkpoints.sh └── requirements.txt ├── .codecov.yml ├── .git-blame-ignore-revs ├── .github ├── BECOMING_A_CORE_CONTRIBUTOR.md ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE │ ├── 1_bug_report.yaml │ ├── 2_refactor.yaml │ ├── 3_feature_request.yaml │ ├── 4_documentation.yaml │ └── config.yml ├── PULL_REQUEST_TEMPLATE.md ├── actions │ ├── pkg-check │ │ └── action.yml │ ├── pkg-install │ │ └── action.yml │ └── pkg-publish │ │ └── action.yml ├── advanced-issue-labeler.yml ├── checkgroup.yml ├── dependabot.yml ├── label-change.yml ├── lightning-probot.yml ├── markdown-links-config.json ├── stale.yml └── workflows │ ├── README.md │ ├── _build-packages.yml │ ├── _legacy-checkpoints.yml │ ├── call-clear-cache.yml │ ├── ci-check-md-links.yml │ ├── ci-checkpoints.yml │ ├── ci-pkg-install.yml │ ├── ci-rtfd.yml │ ├── ci-schema.yml │ ├── ci-tests-fabric.yml │ ├── ci-tests-pytorch.yml │ ├── cleanup-caches.yml │ ├── code-checks.yml │ ├── docs-build.yml │ ├── docs-tutorials.yml │ ├── label-conflicts.yml │ ├── labeler-issue.yml │ ├── labeler-pr.yml │ ├── probot-auto-cc.yml │ ├── probot-check-group.yml │ ├── release-nightly.yml │ ├── release-pkg.yml │ └── tpu-tests.yml.disabled ├── .gitignore ├── .gitmodules ├── .lightning └── workflows │ ├── benchmark.yml │ ├── fabric.yml │ └── pytorch.yml ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── CITATION.cff ├── LICENSE ├── Makefile ├── README.md ├── SECURITY.md ├── docs ├── README.md ├── crossroad.html ├── generate_docs_for_tags.sh ├── redirect.html ├── rtfd-build.sh ├── source-fabric │ ├── Makefile │ ├── _static │ │ ├── copybutton.js │ │ ├── images │ │ │ ├── icon.svg │ │ │ ├── logo-large.svg │ │ │ ├── logo-small.svg │ │ │ ├── logo.png │ │ │ └── logo.svg │ │ └── main.css │ ├── _templates │ │ ├── autosummary │ │ │ └── module.rst │ │ ├── classtemplate.rst │ │ ├── classtemplate_no_index.rst │ │ ├── layout.html │ │ └── theme_variables.jinja │ ├── advanced │ │ ├── compile.rst │ │ ├── distributed_communication.rst │ │ ├── gradient_accumulation.rst │ │ ├── model_init.rst │ │ ├── model_parallel │ │ │ ├── fsdp.rst │ │ │ ├── index.rst │ │ │ ├── tp.rst │ │ │ └── tp_fsdp.rst │ │ └── multiple_setup.rst │ ├── api │ │ ├── accelerators.rst │ │ ├── collectives.rst │ │ ├── environments.rst │ │ ├── fabric.rst │ │ ├── fabric_args.rst │ │ ├── fabric_methods.rst │ │ ├── io.rst │ │ ├── loggers.rst │ │ ├── precision.rst │ │ ├── strategies.rst │ │ ├── utilities.rst │ │ └── wrappers.rst │ ├── conf.py │ ├── examples │ │ └── index.rst │ ├── fundamentals │ │ ├── accelerators.rst │ │ ├── code_structure.rst │ │ ├── convert.rst │ │ ├── installation.rst │ │ ├── launch.rst │ │ ├── notebooks.rst │ │ └── precision.rst │ ├── glossary │ │ └── index.rst │ ├── guide │ │ ├── callbacks.rst │ │ ├── checkpoint │ │ │ ├── checkpoint.rst │ │ │ ├── distributed_checkpoint.rst │ │ │ └── index.rst │ │ ├── index.rst │ │ ├── lightning_module.rst │ │ ├── loggers │ │ │ └── wandb.rst │ │ ├── logging.rst │ │ ├── multi_node │ │ │ ├── barebones.rst │ │ │ ├── cloud.rst │ │ │ ├── other.rst │ │ │ └── slurm.rst │ │ └── trainer_template.rst │ ├── index.rst │ ├── levels │ │ ├── advanced.rst │ │ ├── basic.rst │ │ └── intermediate.rst │ ├── links.rst │ └── make.bat └── source-pytorch │ ├── Makefile │ ├── _static │ ├── copybutton.js │ ├── images │ │ ├── accelerator │ │ │ └── ipus │ │ │ │ └── profiler.png │ │ ├── benchmarks │ │ │ └── figure-parity-times.png │ │ ├── general │ │ │ ├── PTL101_youtube_thumbnail.jpg │ │ │ ├── fast_2.gif │ │ │ ├── pl_overview.gif │ │ │ ├── pl_overview_flat.jpg │ │ │ ├── pl_quick_start_full_compressed.gif │ │ │ ├── tf_loss.jpg │ │ │ ├── tf_tags.jpg │ │ │ └── tutorial_cover.jpg │ │ ├── icon.svg │ │ ├── lightning_lite │ │ │ └── lite.gif │ │ ├── lightning_module │ │ │ ├── pt_to_pl.png │ │ │ └── pt_trainer.png │ │ ├── logo.png │ │ ├── logo.svg │ │ ├── logo_light.svg │ │ ├── mnist_imgs │ │ │ ├── mnist_cpu_bar.png │ │ │ ├── mnist_gpu.png │ │ │ ├── mnist_tb.png │ │ │ ├── pt_to_pl.jpg │ │ │ ├── restart_runtime.png │ │ │ ├── runtime_tpu.png │ │ │ ├── tpu_fast.png │ │ │ └── tpu_start.png │ │ └── trainer │ │ │ └── lr_finder.png │ └── main.css │ ├── _templates │ ├── autosummary │ │ └── module.rst │ ├── classtemplate.rst │ ├── layout.html │ └── theme_variables.jinja │ ├── accelerators │ ├── accelerator_prepare.rst │ ├── gpu.rst │ ├── gpu_advanced.rst │ ├── gpu_basic.rst │ ├── gpu_expert.rst │ ├── gpu_faq.rst │ ├── gpu_intermediate.rst │ ├── mps.rst │ ├── mps_basic.rst │ ├── tpu.rst │ ├── tpu_advanced.rst │ ├── tpu_basic.rst │ ├── tpu_faq.rst │ └── tpu_intermediate.rst │ ├── advanced │ ├── compile.rst │ ├── ddp_optimizations.rst │ ├── finetuning.rst │ ├── model_init.rst │ ├── model_parallel.rst │ ├── model_parallel │ │ ├── deepspeed.rst │ │ ├── fsdp.rst │ │ ├── index.rst │ │ ├── tp.rst │ │ └── tp_fsdp.rst │ ├── post_training_quantization.rst │ ├── pretrained.rst │ ├── pruning_quantization.rst │ ├── speed.rst │ ├── strategy_registry.rst │ ├── training_tricks.rst │ ├── transfer_learning.rst │ └── warnings.rst │ ├── api_references.rst │ ├── benchmarking │ └── benchmarks.rst │ ├── cli │ ├── lightning_cli.rst │ ├── lightning_cli_advanced.rst │ ├── lightning_cli_advanced_2.rst │ ├── lightning_cli_advanced_3.rst │ ├── lightning_cli_expert.rst │ ├── lightning_cli_faq.rst │ ├── lightning_cli_intermediate.rst │ └── lightning_cli_intermediate_2.rst │ ├── clouds │ ├── cluster.rst │ ├── cluster_advanced.rst │ ├── cluster_expert.rst │ ├── cluster_intermediate_1.rst │ ├── cluster_intermediate_2.rst │ └── lightning_ai.rst │ ├── common │ ├── checkpointing.rst │ ├── checkpointing_advanced.rst │ ├── checkpointing_basic.rst │ ├── checkpointing_expert.rst │ ├── checkpointing_intermediate.rst │ ├── checkpointing_migration.rst │ ├── child_modules.rst │ ├── console_logs.rst │ ├── early_stopping.rst │ ├── evaluation.rst │ ├── evaluation_basic.rst │ ├── evaluation_intermediate.rst │ ├── gradient_accumulation.rst │ ├── hooks.rst │ ├── hyperparameters.rst │ ├── index.rst │ ├── lightning_module.rst │ ├── notebooks.rst │ ├── optimization.rst │ ├── precision.rst │ ├── precision_basic.rst │ ├── precision_expert.rst │ ├── precision_intermediate.rst │ ├── progress_bar.rst │ ├── remote_fs.rst │ ├── tbptt.rst │ └── trainer.rst │ ├── common_usecases.rst │ ├── community │ ├── governance.rst │ └── index.rst │ ├── conf.py │ ├── data │ ├── access.rst │ ├── alternatives.rst │ ├── data.rst │ ├── datamodule.rst │ └── iterables.rst │ ├── debug │ ├── debugging.rst │ ├── debugging_advanced.rst │ ├── debugging_basic.rst │ └── debugging_intermediate.rst │ ├── deploy │ ├── production.rst │ ├── production_advanced.rst │ ├── production_advanced_2.rst │ ├── production_basic.rst │ └── production_intermediate.rst │ ├── expertise_levels.rst │ ├── extensions │ ├── accelerator.rst │ ├── callbacks.rst │ ├── callbacks_state.rst │ ├── datamodules_state.rst │ ├── entry_points.rst │ ├── logging.rst │ ├── plugins.rst │ └── strategy.rst │ ├── glossary │ └── index.rst │ ├── index.rst │ ├── integrations │ └── strategies │ │ └── index.rst │ ├── levels │ ├── advanced.rst │ ├── advanced_level_15.rst │ ├── advanced_level_16.rst │ ├── advanced_level_17.rst │ ├── advanced_level_18.rst │ ├── advanced_level_20.rst │ ├── advanced_level_21.rst │ ├── basic_level_2.rst │ ├── basic_level_5.rst │ ├── core_level_3.rst │ ├── core_level_6.rst │ ├── core_skills.rst │ ├── expert.rst │ ├── expert_level_22.rst │ ├── expert_level_23.rst │ ├── expert_level_24.rst │ ├── intermediate.rst │ ├── intermediate_level_10.rst │ ├── intermediate_level_11.rst │ ├── intermediate_level_12.rst │ ├── intermediate_level_13.rst │ ├── intermediate_level_14.rst │ ├── intermediate_level_7.rst │ └── intermediate_level_9.rst │ ├── links.rst │ ├── make.bat │ ├── model │ ├── build_model.rst │ ├── build_model_advanced.rst │ ├── build_model_intermediate.rst │ ├── manual_optimization.rst │ ├── own_your_loop.rst │ └── train_model_basic.rst │ ├── notebooks.rst │ ├── past_versions.rst │ ├── starter │ ├── converting.rst │ ├── installation.rst │ ├── introduction.rst │ └── style_guide.rst │ ├── tuning │ ├── profiler.rst │ ├── profiler_advanced.rst │ ├── profiler_basic.rst │ ├── profiler_expert.rst │ └── profiler_intermediate.rst │ ├── tutorials.rst │ ├── upgrade │ ├── from_1_4.rst │ ├── from_1_5.rst │ ├── from_1_6.rst │ ├── from_1_7.rst │ ├── from_1_8.rst │ ├── from_1_9.rst │ ├── from_2_0.rst │ ├── migration_guide.rst │ └── sections │ │ ├── 1_4_advanced.rst │ │ ├── 1_4_devel.rst │ │ ├── 1_4_regular.rst │ │ ├── 1_5_advanced.rst │ │ ├── 1_5_devel.rst │ │ ├── 1_5_regular.rst │ │ ├── 1_6_advanced.rst │ │ ├── 1_6_devel.rst │ │ ├── 1_6_regular.rst │ │ ├── 1_7_advanced.rst │ │ ├── 1_7_devel.rst │ │ ├── 1_7_regular.rst │ │ ├── 1_8_advanced.rst │ │ ├── 1_8_devel.rst │ │ ├── 1_8_regular.rst │ │ ├── 1_9_advanced.rst │ │ ├── 1_9_devel.rst │ │ ├── 1_9_regular.rst │ │ ├── 2_0_advanced.rst │ │ ├── 2_0_devel.rst │ │ └── 2_0_regular.rst │ ├── versioning.rst │ └── visualize │ ├── experiment_managers.rst │ ├── loggers.rst │ ├── logging_advanced.rst │ ├── logging_basic.rst │ ├── logging_expert.rst │ ├── logging_intermediate.rst │ └── supported_exp_managers.rst ├── examples ├── README.md ├── fabric │ ├── build_your_own_trainer │ │ ├── README.md │ │ ├── run.py │ │ └── trainer.py │ ├── dcgan │ │ ├── README.md │ │ ├── train_fabric.py │ │ └── train_torch.py │ ├── fp8_distributed_transformer │ │ ├── README.md │ │ ├── requirements.txt │ │ └── train.py │ ├── image_classifier │ │ ├── README.md │ │ ├── train_fabric.py │ │ └── train_torch.py │ ├── kfold_cv │ │ ├── README.md │ │ └── train_fabric.py │ ├── language_model │ │ ├── README.md │ │ └── train.py │ ├── meta_learning │ │ ├── README.md │ │ ├── train_fabric.py │ │ └── train_torch.py │ ├── reinforcement_learning │ │ ├── README.md │ │ ├── requirements.txt │ │ ├── rl │ │ │ ├── __init__.py │ │ │ ├── agent.py │ │ │ ├── loss.py │ │ │ └── utils.py │ │ ├── train_fabric.py │ │ ├── train_fabric_decoupled.py │ │ └── train_torch.py │ └── tensor_parallel │ │ ├── README.md │ │ ├── data.py │ │ ├── model.py │ │ ├── parallelism.py │ │ └── train.py ├── pytorch │ ├── basics │ │ ├── README.md │ │ ├── autoencoder.py │ │ ├── backbone_image_classifier.py │ │ ├── profiler_example.py │ │ └── transformer.py │ ├── bug_report │ │ ├── bug_report_model.ipynb │ │ └── bug_report_model.py │ ├── domain_templates │ │ ├── computer_vision_fine_tuning.py │ │ ├── generative_adversarial_net.py │ │ ├── imagenet.py │ │ ├── reinforce_learn_Qnet.py │ │ ├── reinforce_learn_ppo.py │ │ └── semantic_segmentation.py │ ├── fp8_distributed_transformer │ │ ├── README.md │ │ ├── requirements.txt │ │ └── train.py │ ├── servable_module │ │ └── production.py │ └── tensor_parallel │ │ ├── README.md │ │ ├── data.py │ │ ├── model.py │ │ ├── parallelism.py │ │ └── train.py ├── run_fabric_examples.sh └── run_pl_examples.sh ├── pyproject.toml ├── requirements.txt ├── requirements ├── README.md ├── ci.txt ├── collect_env_details.py ├── data │ └── data.txt ├── docs.txt ├── doctests.txt ├── fabric │ ├── base.txt │ ├── docs.txt │ ├── examples.txt │ ├── extra.txt │ ├── strategies.txt │ └── test.txt ├── pytorch │ ├── base.txt │ ├── check-avail-extras.py │ ├── docs.txt │ ├── examples.txt │ ├── extra.txt │ ├── loggers.info │ ├── strategies.txt │ └── test.txt └── typing.txt ├── setup.py ├── src ├── lightning │ ├── MANIFEST.in │ ├── __about__.py │ ├── __init__.py │ ├── __setup__.py │ ├── __version__.py │ ├── data │ │ ├── README.md │ │ └── __init__.py │ ├── fabric │ │ ├── CHANGELOG.md │ │ ├── __init__.py │ │ ├── _graveyard │ │ │ ├── __init__.py │ │ │ └── tpu.py │ │ ├── accelerators │ │ │ ├── __init__.py │ │ │ ├── accelerator.py │ │ │ ├── cpu.py │ │ │ ├── cuda.py │ │ │ ├── mps.py │ │ │ ├── registry.py │ │ │ └── xla.py │ │ ├── cli.py │ │ ├── connector.py │ │ ├── fabric.py │ │ ├── loggers │ │ │ ├── __init__.py │ │ │ ├── csv_logs.py │ │ │ ├── logger.py │ │ │ └── tensorboard.py │ │ ├── plugins │ │ │ ├── __init__.py │ │ │ ├── collectives │ │ │ │ ├── __init__.py │ │ │ │ ├── collective.py │ │ │ │ ├── single_device.py │ │ │ │ └── torch_collective.py │ │ │ ├── environments │ │ │ │ ├── __init__.py │ │ │ │ ├── cluster_environment.py │ │ │ │ ├── kubeflow.py │ │ │ │ ├── lightning.py │ │ │ │ ├── lsf.py │ │ │ │ ├── mpi.py │ │ │ │ ├── slurm.py │ │ │ │ ├── torchelastic.py │ │ │ │ └── xla.py │ │ │ ├── io │ │ │ │ ├── __init__.py │ │ │ │ ├── checkpoint_io.py │ │ │ │ ├── torch_io.py │ │ │ │ └── xla.py │ │ │ └── precision │ │ │ │ ├── __init__.py │ │ │ │ ├── amp.py │ │ │ │ ├── bitsandbytes.py │ │ │ │ ├── deepspeed.py │ │ │ │ ├── double.py │ │ │ │ ├── fsdp.py │ │ │ │ ├── half.py │ │ │ │ ├── precision.py │ │ │ │ ├── transformer_engine.py │ │ │ │ ├── utils.py │ │ │ │ └── xla.py │ │ ├── strategies │ │ │ ├── __init__.py │ │ │ ├── ddp.py │ │ │ ├── deepspeed.py │ │ │ ├── dp.py │ │ │ ├── fsdp.py │ │ │ ├── launchers │ │ │ │ ├── __init__.py │ │ │ │ ├── launcher.py │ │ │ │ ├── multiprocessing.py │ │ │ │ ├── subprocess_script.py │ │ │ │ └── xla.py │ │ │ ├── model_parallel.py │ │ │ ├── parallel.py │ │ │ ├── registry.py │ │ │ ├── single_device.py │ │ │ ├── single_xla.py │ │ │ ├── strategy.py │ │ │ ├── xla.py │ │ │ └── xla_fsdp.py │ │ ├── utilities │ │ │ ├── __init__.py │ │ │ ├── apply_func.py │ │ │ ├── cloud_io.py │ │ │ ├── consolidate_checkpoint.py │ │ │ ├── data.py │ │ │ ├── device_dtype_mixin.py │ │ │ ├── device_parser.py │ │ │ ├── distributed.py │ │ │ ├── enums.py │ │ │ ├── exceptions.py │ │ │ ├── imports.py │ │ │ ├── init.py │ │ │ ├── load.py │ │ │ ├── logger.py │ │ │ ├── optimizer.py │ │ │ ├── rank_zero.py │ │ │ ├── registry.py │ │ │ ├── seed.py │ │ │ ├── spike.py │ │ │ ├── testing │ │ │ │ ├── __init__.py │ │ │ │ └── _runif.py │ │ │ ├── throughput.py │ │ │ ├── types.py │ │ │ └── warnings.py │ │ └── wrappers.py │ ├── py.typed │ └── pytorch │ │ ├── CHANGELOG.md │ │ ├── __init__.py │ │ ├── _graveyard │ │ ├── __init__.py │ │ ├── _torchmetrics.py │ │ ├── hpu.py │ │ ├── precision.py │ │ └── tpu.py │ │ ├── accelerators │ │ ├── __init__.py │ │ ├── accelerator.py │ │ ├── cpu.py │ │ ├── cuda.py │ │ ├── mps.py │ │ └── xla.py │ │ ├── callbacks │ │ ├── __init__.py │ │ ├── batch_size_finder.py │ │ ├── callback.py │ │ ├── checkpoint.py │ │ ├── device_stats_monitor.py │ │ ├── early_stopping.py │ │ ├── finetuning.py │ │ ├── gradient_accumulation_scheduler.py │ │ ├── lambda_function.py │ │ ├── lr_finder.py │ │ ├── lr_monitor.py │ │ ├── model_checkpoint.py │ │ ├── model_summary.py │ │ ├── on_exception_checkpoint.py │ │ ├── prediction_writer.py │ │ ├── progress │ │ │ ├── __init__.py │ │ │ ├── progress_bar.py │ │ │ ├── rich_progress.py │ │ │ └── tqdm_progress.py │ │ ├── pruning.py │ │ ├── rich_model_summary.py │ │ ├── spike.py │ │ ├── stochastic_weight_avg.py │ │ ├── throughput_monitor.py │ │ ├── timer.py │ │ └── weight_averaging.py │ │ ├── cli.py │ │ ├── core │ │ ├── __init__.py │ │ ├── datamodule.py │ │ ├── hooks.py │ │ ├── mixins │ │ │ ├── __init__.py │ │ │ └── hparams_mixin.py │ │ ├── module.py │ │ ├── optimizer.py │ │ └── saving.py │ │ ├── demos │ │ ├── __init__.py │ │ ├── boring_classes.py │ │ ├── lstm.py │ │ ├── mnist_datamodule.py │ │ └── transformer.py │ │ ├── loggers │ │ ├── __init__.py │ │ ├── comet.py │ │ ├── csv_logs.py │ │ ├── logger.py │ │ ├── mlflow.py │ │ ├── neptune.py │ │ ├── tensorboard.py │ │ ├── utilities.py │ │ └── wandb.py │ │ ├── loops │ │ ├── __init__.py │ │ ├── evaluation_loop.py │ │ ├── fetchers.py │ │ ├── fit_loop.py │ │ ├── loop.py │ │ ├── optimization │ │ │ ├── __init__.py │ │ │ ├── automatic.py │ │ │ ├── closure.py │ │ │ └── manual.py │ │ ├── prediction_loop.py │ │ ├── progress.py │ │ ├── training_epoch_loop.py │ │ └── utilities.py │ │ ├── overrides │ │ ├── __init__.py │ │ └── distributed.py │ │ ├── plugins │ │ ├── __init__.py │ │ ├── environments │ │ │ └── __init__.py │ │ ├── io │ │ │ ├── __init__.py │ │ │ ├── async_plugin.py │ │ │ ├── checkpoint_plugin.py │ │ │ ├── torch_plugin.py │ │ │ ├── wrapper.py │ │ │ └── xla_plugin.py │ │ ├── layer_sync.py │ │ └── precision │ │ │ ├── __init__.py │ │ │ ├── amp.py │ │ │ ├── bitsandbytes.py │ │ │ ├── deepspeed.py │ │ │ ├── double.py │ │ │ ├── fsdp.py │ │ │ ├── half.py │ │ │ ├── precision.py │ │ │ ├── transformer_engine.py │ │ │ └── xla.py │ │ ├── profilers │ │ ├── __init__.py │ │ ├── advanced.py │ │ ├── base.py │ │ ├── profiler.py │ │ ├── pytorch.py │ │ ├── simple.py │ │ └── xla.py │ │ ├── serve │ │ ├── __init__.py │ │ ├── servable_module.py │ │ └── servable_module_validator.py │ │ ├── strategies │ │ ├── __init__.py │ │ ├── ddp.py │ │ ├── deepspeed.py │ │ ├── fsdp.py │ │ ├── launchers │ │ │ ├── __init__.py │ │ │ ├── launcher.py │ │ │ ├── multiprocessing.py │ │ │ ├── subprocess_script.py │ │ │ └── xla.py │ │ ├── model_parallel.py │ │ ├── parallel.py │ │ ├── single_device.py │ │ ├── single_xla.py │ │ ├── strategy.py │ │ └── xla.py │ │ ├── trainer │ │ ├── __init__.py │ │ ├── call.py │ │ ├── configuration_validator.py │ │ ├── connectors │ │ │ ├── __init__.py │ │ │ ├── accelerator_connector.py │ │ │ ├── callback_connector.py │ │ │ ├── checkpoint_connector.py │ │ │ ├── data_connector.py │ │ │ ├── logger_connector │ │ │ │ ├── __init__.py │ │ │ │ ├── fx_validator.py │ │ │ │ ├── logger_connector.py │ │ │ │ └── result.py │ │ │ └── signal_connector.py │ │ ├── setup.py │ │ ├── states.py │ │ └── trainer.py │ │ ├── tuner │ │ ├── __init__.py │ │ ├── batch_size_scaling.py │ │ ├── lr_finder.py │ │ └── tuning.py │ │ └── utilities │ │ ├── __init__.py │ │ ├── _pytree.py │ │ ├── argparse.py │ │ ├── combined_loader.py │ │ ├── compile.py │ │ ├── consolidate_checkpoint.py │ │ ├── data.py │ │ ├── deepspeed.py │ │ ├── enums.py │ │ ├── exceptions.py │ │ ├── grads.py │ │ ├── imports.py │ │ ├── memory.py │ │ ├── migration │ │ ├── __init__.py │ │ ├── migration.py │ │ └── utils.py │ │ ├── model_helpers.py │ │ ├── model_registry.py │ │ ├── model_summary │ │ ├── __init__.py │ │ ├── model_summary.py │ │ └── model_summary_deepspeed.py │ │ ├── parameter_tying.py │ │ ├── parsing.py │ │ ├── rank_zero.py │ │ ├── seed.py │ │ ├── signature_utils.py │ │ ├── testing │ │ ├── __init__.py │ │ └── _runif.py │ │ ├── types.py │ │ ├── upgrade_checkpoint.py │ │ └── warnings.py ├── lightning_fabric │ ├── MANIFEST.in │ ├── README.md │ ├── __about__.py │ ├── __setup__.py │ ├── __version__.py │ ├── py.typed │ └── shell-folder_code-lives-lightning.info ├── pytorch_lightning │ ├── MANIFEST.in │ ├── README.md │ ├── __about__.py │ ├── __setup__.py │ ├── __version__.py │ ├── py.typed │ └── shell-folder_code-lives-lightning.info └── version.info └── tests ├── README.md ├── legacy ├── README.md ├── back-compatible-versions.txt ├── checkpoints │ └── .gitkeep ├── generate_checkpoints.sh └── simple_classif_training.py ├── parity_fabric ├── __init__.py ├── conftest.py ├── models.py ├── run_standalone_tasks.sh ├── test_parity_ddp.py ├── test_parity_simple.py └── utils.py ├── parity_pytorch ├── __init__.py ├── generate_comparison.py ├── measure.py ├── models.py ├── test_basic_parity.py └── test_sync_batchnorm_parity.py ├── tests_fabric ├── __init__.py ├── accelerators │ ├── __init__.py │ ├── test_cpu.py │ ├── test_cuda.py │ ├── test_mps.py │ ├── test_registry.py │ └── test_xla.py ├── conftest.py ├── graveyard │ └── test_tpu.py ├── helpers │ ├── __init__.py │ ├── dataloaders.py │ ├── datasets.py │ └── runif.py ├── loggers │ ├── __init__.py │ ├── test_csv.py │ └── test_tensorboard.py ├── plugins │ ├── __init__.py │ ├── collectives │ │ ├── __init__.py │ │ ├── test_single_device.py │ │ └── test_torch_collective.py │ ├── environments │ │ ├── __init__.py │ │ ├── test_kubeflow.py │ │ ├── test_lightning.py │ │ ├── test_lsf.py │ │ ├── test_mpi.py │ │ ├── test_slurm.py │ │ ├── test_torchelastic.py │ │ └── test_xla.py │ └── precision │ │ ├── __init__.py │ │ ├── test_all.py │ │ ├── test_amp.py │ │ ├── test_amp_integration.py │ │ ├── test_bitsandbytes.py │ │ ├── test_deepspeed.py │ │ ├── test_deepspeed_integration.py │ │ ├── test_double.py │ │ ├── test_double_integration.py │ │ ├── test_fsdp.py │ │ ├── test_half.py │ │ ├── test_transformer_engine.py │ │ ├── test_utils.py │ │ ├── test_xla.py │ │ └── test_xla_integration.py ├── run_tpu_tests.sh ├── strategies │ ├── __init__.py │ ├── launchers │ │ ├── __init__.py │ │ ├── test_multiprocessing.py │ │ ├── test_multiprocessing_integration.py │ │ └── test_subprocess_script.py │ ├── test_ddp.py │ ├── test_ddp_integration.py │ ├── test_deepspeed.py │ ├── test_deepspeed_integration.py │ ├── test_dp.py │ ├── test_fsdp.py │ ├── test_fsdp_integration.py │ ├── test_model_parallel.py │ ├── test_model_parallel_integration.py │ ├── test_registry.py │ ├── test_single_device.py │ ├── test_strategy.py │ ├── test_xla.py │ ├── test_xla_fsdp.py │ └── test_xla_fsdp_integration.py ├── test_cli.py ├── test_connector.py ├── test_fabric.py ├── test_wrappers.py └── utilities │ ├── __init__.py │ ├── test_apply_func.py │ ├── test_cloud_io.py │ ├── test_consolidate_checkpoint.py │ ├── test_data.py │ ├── test_device_dtype_mixin.py │ ├── test_device_parser.py │ ├── test_distributed.py │ ├── test_imports.py │ ├── test_init.py │ ├── test_load.py │ ├── test_logger.py │ ├── test_optimizer.py │ ├── test_rank_zero.py │ ├── test_registry.py │ ├── test_runif.py │ ├── test_seed.py │ ├── test_spike.py │ ├── test_throughput.py │ └── test_warnings.py └── tests_pytorch ├── __init__.py ├── accelerators ├── __init__.py ├── test_common.py ├── test_cpu.py ├── test_gpu.py ├── test_mps.py ├── test_registry.py └── test_xla.py ├── callbacks ├── __init__.py ├── progress │ ├── test_rich_progress_bar.py │ └── test_tqdm_progress_bar.py ├── test_callback_hook_outputs.py ├── test_callbacks.py ├── test_device_stats_monitor.py ├── test_early_stopping.py ├── test_finetuning_callback.py ├── test_gradient_accumulation_scheduler.py ├── test_lambda_function.py ├── test_lr_monitor.py ├── test_model_checkpoint_additional_cases.py ├── test_model_checkpoint_edge_cases.py ├── test_model_checkpoint_manual_opt.py ├── test_model_checkpoint_step_interval_val_metric.py ├── test_model_summary.py ├── test_prediction_writer.py ├── test_pruning.py ├── test_rich_model_summary.py ├── test_spike.py ├── test_stochastic_weight_avg.py ├── test_throughput_monitor.py ├── test_timer.py └── test_weight_averaging.py ├── checkpointing ├── __init__.py ├── test_checkpoint_callback_frequency.py ├── test_legacy_checkpoints.py ├── test_model_checkpoint.py ├── test_torch_saving.py └── test_trainer_checkpoint.py ├── conftest.py ├── core ├── __init__.py ├── test_datamodules.py ├── test_lightning_module.py ├── test_lightning_optimizer.py ├── test_metric_result_integration.py ├── test_results.py └── test_saving.py ├── demos ├── __init__.py ├── lstm.py └── transformer.py ├── deprecated_api ├── __init__.py └── test_no_removal_version.py ├── graveyard ├── __init__.py ├── test_hpu.py ├── test_precision.py └── test_tpu.py ├── helpers ├── __init__.py ├── advanced_models.py ├── dataloaders.py ├── datamodules.py ├── datasets.py ├── deterministic_model.py ├── pipelines.py ├── runif.py ├── simple_models.py ├── test_datasets.py ├── test_models.py ├── threading.py └── utils.py ├── loggers ├── __init__.py ├── conftest.py ├── test_all.py ├── test_comet.py ├── test_csv.py ├── test_logger.py ├── test_mlflow.py ├── test_neptune.py ├── test_tensorboard.py ├── test_utilities.py └── test_wandb.py ├── loops ├── __init__.py ├── optimization │ ├── __init__.py │ ├── test_automatic_loop.py │ ├── test_closure.py │ └── test_manual_loop.py ├── test_all.py ├── test_double_iter_in_iterable_dataset.py ├── test_evaluation_loop.py ├── test_evaluation_loop_flow.py ├── test_fetchers.py ├── test_flow_warnings.py ├── test_loop_state_dict.py ├── test_loops.py ├── test_prediction_loop.py ├── test_progress.py ├── test_training_epoch_loop.py ├── test_training_loop.py ├── test_training_loop_flow_dict.py └── test_training_loop_flow_scalar.py ├── models ├── __init__.py ├── conf │ ├── config.yaml │ └── training │ │ └── default.yaml ├── test_amp.py ├── test_cpu.py ├── test_ddp_fork_amp.py ├── test_fabric_integration.py ├── test_gpu.py ├── test_hooks.py ├── test_hparams.py ├── test_onnx.py ├── test_restore.py ├── test_torch_tensorrt.py ├── test_torchscript.py └── test_tpu.py ├── overrides ├── __init__.py └── test_distributed.py ├── plugins ├── __init__.py ├── precision │ ├── __init__.py │ ├── test_all.py │ ├── test_amp.py │ ├── test_amp_integration.py │ ├── test_bitsandbytes.py │ ├── test_deepspeed_precision.py │ ├── test_double.py │ ├── test_fsdp.py │ ├── test_half.py │ ├── test_transformer_engine.py │ └── test_xla.py ├── test_amp_plugins.py ├── test_async_checkpoint.py ├── test_checkpoint_io_plugin.py └── test_cluster_integration.py ├── profilers ├── __init__.py ├── test_profiler.py └── test_xla_profiler.py ├── run_standalone_tasks.sh ├── run_tpu_tests.sh ├── serve ├── __init__.py └── test_servable_module_validator.py ├── strategies ├── __init__.py ├── launchers │ ├── __init__.py │ ├── test_multiprocessing.py │ └── test_subprocess_script.py ├── scripts │ ├── __init__.py │ └── cli_script.py ├── test_common.py ├── test_custom_strategy.py ├── test_ddp.py ├── test_ddp_integration.py ├── test_ddp_integration_comm_hook.py ├── test_deepspeed.py ├── test_fsdp.py ├── test_model_parallel.py ├── test_model_parallel_integration.py ├── test_registry.py ├── test_single_device.py └── test_xla.py ├── test_cli.py ├── trainer ├── __init__.py ├── connectors │ ├── __init__.py │ ├── test_accelerator_connector.py │ ├── test_callback_connector.py │ ├── test_checkpoint_connector.py │ ├── test_data_connector.py │ ├── test_logger_connector.py │ ├── test_rich_integration.py │ └── test_signal_connector.py ├── dynamic_args │ ├── __init__.py │ └── test_multiple_eval_dataloaders.py ├── flags │ ├── __init__.py │ ├── test_barebones.py │ ├── test_check_val_every_n_epoch.py │ ├── test_env_vars.py │ ├── test_fast_dev_run.py │ ├── test_inference_mode.py │ ├── test_limit_batches.py │ ├── test_min_max_epochs.py │ ├── test_overfit_batches.py │ └── test_val_check_interval.py ├── logging_ │ ├── __init__.py │ ├── test_distributed_logging.py │ ├── test_eval_loop_logging.py │ ├── test_logger_connector.py │ ├── test_loop_logging.py │ └── test_train_loop_logging.py ├── optimization │ ├── __init__.py │ ├── test_backward_calls.py │ ├── test_manual_optimization.py │ ├── test_multiple_optimizers.py │ └── test_optimizers.py ├── properties │ ├── __init__.py │ ├── test_estimated_stepping_batches.py │ ├── test_get_model.py │ ├── test_log_dir.py │ └── test_loggers.py ├── test_config_validator.py ├── test_dataloaders.py ├── test_ddp_sigterm_handling.py ├── test_states.py └── test_trainer.py ├── tuner ├── __init__.py ├── test_lr_finder.py ├── test_scale_batch_size.py └── test_tuning.py └── utilities ├── __init__.py ├── migration ├── __init__.py ├── test_migration.py └── test_utils.py ├── test_all_gather_grad.py ├── test_auto_restart.py ├── test_combined_loader.py ├── test_compile.py ├── test_consolidate_checkpoint.py ├── test_data.py ├── test_deepspeed_collate_checkpoint.py ├── test_deepspeed_model_summary.py ├── test_dtype_device_mixin.py ├── test_enums.py ├── test_exceptions.py ├── test_grads.py ├── test_imports.py ├── test_memory.py ├── test_model_helpers.py ├── test_model_summary.py ├── test_parameter_tying.py ├── test_parsing.py ├── test_pytree.py ├── test_seed.py ├── test_signature_utils.py ├── test_types.py ├── test_upgrade_checkpoint.py ├── test_warnings.py └── testing ├── __init__.py └── test_runif.py /.actions/assistant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.actions/assistant.py -------------------------------------------------------------------------------- /.actions/pull_legacy_checkpoints.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.actions/pull_legacy_checkpoints.sh -------------------------------------------------------------------------------- /.actions/requirements.txt: -------------------------------------------------------------------------------- 1 | jsonargparse 2 | requests 3 | packaging 4 | -------------------------------------------------------------------------------- /.codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.codecov.yml -------------------------------------------------------------------------------- /.git-blame-ignore-revs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.git-blame-ignore-revs -------------------------------------------------------------------------------- /.github/BECOMING_A_CORE_CONTRIBUTOR.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/BECOMING_A_CORE_CONTRIBUTOR.md -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/1_bug_report.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/ISSUE_TEMPLATE/1_bug_report.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/2_refactor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/ISSUE_TEMPLATE/2_refactor.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/3_feature_request.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/ISSUE_TEMPLATE/3_feature_request.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/4_documentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/ISSUE_TEMPLATE/4_documentation.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/actions/pkg-check/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/actions/pkg-check/action.yml -------------------------------------------------------------------------------- /.github/actions/pkg-install/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/actions/pkg-install/action.yml -------------------------------------------------------------------------------- /.github/actions/pkg-publish/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/actions/pkg-publish/action.yml -------------------------------------------------------------------------------- /.github/advanced-issue-labeler.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/advanced-issue-labeler.yml -------------------------------------------------------------------------------- /.github/checkgroup.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/checkgroup.yml -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/label-change.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/label-change.yml -------------------------------------------------------------------------------- /.github/lightning-probot.yml: -------------------------------------------------------------------------------- 1 | tracking_issue: 10530 2 | -------------------------------------------------------------------------------- /.github/markdown-links-config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/markdown-links-config.json -------------------------------------------------------------------------------- /.github/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/stale.yml -------------------------------------------------------------------------------- /.github/workflows/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/README.md -------------------------------------------------------------------------------- /.github/workflows/_build-packages.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/_build-packages.yml -------------------------------------------------------------------------------- /.github/workflows/_legacy-checkpoints.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/_legacy-checkpoints.yml -------------------------------------------------------------------------------- /.github/workflows/call-clear-cache.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/call-clear-cache.yml -------------------------------------------------------------------------------- /.github/workflows/ci-check-md-links.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/ci-check-md-links.yml -------------------------------------------------------------------------------- /.github/workflows/ci-checkpoints.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/ci-checkpoints.yml -------------------------------------------------------------------------------- /.github/workflows/ci-pkg-install.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/ci-pkg-install.yml -------------------------------------------------------------------------------- /.github/workflows/ci-rtfd.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/ci-rtfd.yml -------------------------------------------------------------------------------- /.github/workflows/ci-schema.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/ci-schema.yml -------------------------------------------------------------------------------- /.github/workflows/ci-tests-fabric.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/ci-tests-fabric.yml -------------------------------------------------------------------------------- /.github/workflows/ci-tests-pytorch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/ci-tests-pytorch.yml -------------------------------------------------------------------------------- /.github/workflows/cleanup-caches.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/cleanup-caches.yml -------------------------------------------------------------------------------- /.github/workflows/code-checks.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/code-checks.yml -------------------------------------------------------------------------------- /.github/workflows/docs-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/docs-build.yml -------------------------------------------------------------------------------- /.github/workflows/docs-tutorials.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/docs-tutorials.yml -------------------------------------------------------------------------------- /.github/workflows/label-conflicts.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/label-conflicts.yml -------------------------------------------------------------------------------- /.github/workflows/labeler-issue.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/labeler-issue.yml -------------------------------------------------------------------------------- /.github/workflows/labeler-pr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/labeler-pr.yml -------------------------------------------------------------------------------- /.github/workflows/probot-auto-cc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/probot-auto-cc.yml -------------------------------------------------------------------------------- /.github/workflows/probot-check-group.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/probot-check-group.yml -------------------------------------------------------------------------------- /.github/workflows/release-nightly.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/release-nightly.yml -------------------------------------------------------------------------------- /.github/workflows/release-pkg.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/release-pkg.yml -------------------------------------------------------------------------------- /.github/workflows/tpu-tests.yml.disabled: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.github/workflows/tpu-tests.yml.disabled -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.gitmodules -------------------------------------------------------------------------------- /.lightning/workflows/benchmark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.lightning/workflows/benchmark.yml -------------------------------------------------------------------------------- /.lightning/workflows/fabric.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.lightning/workflows/fabric.yml -------------------------------------------------------------------------------- /.lightning/workflows/pytorch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.lightning/workflows/pytorch.yml -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- 1 | developer@lightning.ai 2 | -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/crossroad.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/crossroad.html -------------------------------------------------------------------------------- /docs/generate_docs_for_tags.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/generate_docs_for_tags.sh -------------------------------------------------------------------------------- /docs/redirect.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/redirect.html -------------------------------------------------------------------------------- /docs/rtfd-build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/rtfd-build.sh -------------------------------------------------------------------------------- /docs/source-fabric/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/Makefile -------------------------------------------------------------------------------- /docs/source-fabric/_static/copybutton.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_static/copybutton.js -------------------------------------------------------------------------------- /docs/source-fabric/_static/images/icon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_static/images/icon.svg -------------------------------------------------------------------------------- /docs/source-fabric/_static/images/logo-large.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_static/images/logo-large.svg -------------------------------------------------------------------------------- /docs/source-fabric/_static/images/logo-small.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_static/images/logo-small.svg -------------------------------------------------------------------------------- /docs/source-fabric/_static/images/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_static/images/logo.png -------------------------------------------------------------------------------- /docs/source-fabric/_static/images/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_static/images/logo.svg -------------------------------------------------------------------------------- /docs/source-fabric/_static/main.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_static/main.css -------------------------------------------------------------------------------- /docs/source-fabric/_templates/autosummary/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_templates/autosummary/module.rst -------------------------------------------------------------------------------- /docs/source-fabric/_templates/classtemplate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_templates/classtemplate.rst -------------------------------------------------------------------------------- /docs/source-fabric/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_templates/layout.html -------------------------------------------------------------------------------- /docs/source-fabric/_templates/theme_variables.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/_templates/theme_variables.jinja -------------------------------------------------------------------------------- /docs/source-fabric/advanced/compile.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/advanced/compile.rst -------------------------------------------------------------------------------- /docs/source-fabric/advanced/gradient_accumulation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/advanced/gradient_accumulation.rst -------------------------------------------------------------------------------- /docs/source-fabric/advanced/model_init.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/advanced/model_init.rst -------------------------------------------------------------------------------- /docs/source-fabric/advanced/model_parallel/fsdp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/advanced/model_parallel/fsdp.rst -------------------------------------------------------------------------------- /docs/source-fabric/advanced/model_parallel/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/advanced/model_parallel/index.rst -------------------------------------------------------------------------------- /docs/source-fabric/advanced/model_parallel/tp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/advanced/model_parallel/tp.rst -------------------------------------------------------------------------------- /docs/source-fabric/advanced/multiple_setup.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/advanced/multiple_setup.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/accelerators.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/accelerators.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/collectives.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/collectives.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/environments.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/environments.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/fabric.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/fabric.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/fabric_args.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/fabric_args.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/fabric_methods.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/fabric_methods.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/io.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/io.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/loggers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/loggers.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/precision.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/strategies.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/strategies.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/utilities.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/utilities.rst -------------------------------------------------------------------------------- /docs/source-fabric/api/wrappers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/api/wrappers.rst -------------------------------------------------------------------------------- /docs/source-fabric/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/conf.py -------------------------------------------------------------------------------- /docs/source-fabric/examples/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/examples/index.rst -------------------------------------------------------------------------------- /docs/source-fabric/fundamentals/accelerators.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/fundamentals/accelerators.rst -------------------------------------------------------------------------------- /docs/source-fabric/fundamentals/code_structure.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/fundamentals/code_structure.rst -------------------------------------------------------------------------------- /docs/source-fabric/fundamentals/convert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/fundamentals/convert.rst -------------------------------------------------------------------------------- /docs/source-fabric/fundamentals/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/fundamentals/installation.rst -------------------------------------------------------------------------------- /docs/source-fabric/fundamentals/launch.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/fundamentals/launch.rst -------------------------------------------------------------------------------- /docs/source-fabric/fundamentals/notebooks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/fundamentals/notebooks.rst -------------------------------------------------------------------------------- /docs/source-fabric/fundamentals/precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/fundamentals/precision.rst -------------------------------------------------------------------------------- /docs/source-fabric/glossary/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/glossary/index.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/callbacks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/callbacks.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/checkpoint/checkpoint.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/checkpoint/checkpoint.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/checkpoint/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/checkpoint/index.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/index.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/lightning_module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/lightning_module.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/loggers/wandb.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/loggers/wandb.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/logging.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/logging.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/multi_node/barebones.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/multi_node/barebones.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/multi_node/cloud.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/multi_node/cloud.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/multi_node/other.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/multi_node/other.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/multi_node/slurm.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/multi_node/slurm.rst -------------------------------------------------------------------------------- /docs/source-fabric/guide/trainer_template.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/guide/trainer_template.rst -------------------------------------------------------------------------------- /docs/source-fabric/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/index.rst -------------------------------------------------------------------------------- /docs/source-fabric/levels/advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/levels/advanced.rst -------------------------------------------------------------------------------- /docs/source-fabric/levels/basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/levels/basic.rst -------------------------------------------------------------------------------- /docs/source-fabric/levels/intermediate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/levels/intermediate.rst -------------------------------------------------------------------------------- /docs/source-fabric/links.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/links.rst -------------------------------------------------------------------------------- /docs/source-fabric/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-fabric/make.bat -------------------------------------------------------------------------------- /docs/source-pytorch/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/Makefile -------------------------------------------------------------------------------- /docs/source-pytorch/_static/copybutton.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_static/copybutton.js -------------------------------------------------------------------------------- /docs/source-pytorch/_static/images/general/fast_2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_static/images/general/fast_2.gif -------------------------------------------------------------------------------- /docs/source-pytorch/_static/images/icon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_static/images/icon.svg -------------------------------------------------------------------------------- /docs/source-pytorch/_static/images/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_static/images/logo.png -------------------------------------------------------------------------------- /docs/source-pytorch/_static/images/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_static/images/logo.svg -------------------------------------------------------------------------------- /docs/source-pytorch/_static/images/logo_light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_static/images/logo_light.svg -------------------------------------------------------------------------------- /docs/source-pytorch/_static/main.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_static/main.css -------------------------------------------------------------------------------- /docs/source-pytorch/_templates/autosummary/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_templates/autosummary/module.rst -------------------------------------------------------------------------------- /docs/source-pytorch/_templates/classtemplate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_templates/classtemplate.rst -------------------------------------------------------------------------------- /docs/source-pytorch/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_templates/layout.html -------------------------------------------------------------------------------- /docs/source-pytorch/_templates/theme_variables.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/_templates/theme_variables.jinja -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/gpu.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/gpu.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/gpu_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/gpu_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/gpu_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/gpu_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/gpu_expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/gpu_expert.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/gpu_faq.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/gpu_faq.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/gpu_intermediate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/gpu_intermediate.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/mps.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/mps.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/mps_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/mps_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/tpu.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/tpu.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/tpu_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/tpu_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/tpu_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/tpu_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/tpu_faq.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/tpu_faq.rst -------------------------------------------------------------------------------- /docs/source-pytorch/accelerators/tpu_intermediate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/accelerators/tpu_intermediate.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/compile.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/compile.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/ddp_optimizations.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/ddp_optimizations.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/finetuning.rst: -------------------------------------------------------------------------------- 1 | .. include:: transfer_learning.rst 2 | -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/model_init.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/model_init.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/model_parallel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/model_parallel.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/model_parallel/fsdp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/model_parallel/fsdp.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/model_parallel/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/model_parallel/index.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/model_parallel/tp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/model_parallel/tp.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/pretrained.rst: -------------------------------------------------------------------------------- 1 | .. include:: transfer_learning.rst 2 | -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/pruning_quantization.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/pruning_quantization.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/speed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/speed.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/strategy_registry.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/strategy_registry.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/training_tricks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/training_tricks.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/transfer_learning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/transfer_learning.rst -------------------------------------------------------------------------------- /docs/source-pytorch/advanced/warnings.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/advanced/warnings.rst -------------------------------------------------------------------------------- /docs/source-pytorch/api_references.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/api_references.rst -------------------------------------------------------------------------------- /docs/source-pytorch/benchmarking/benchmarks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/benchmarking/benchmarks.rst -------------------------------------------------------------------------------- /docs/source-pytorch/cli/lightning_cli.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/cli/lightning_cli.rst -------------------------------------------------------------------------------- /docs/source-pytorch/cli/lightning_cli_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/cli/lightning_cli_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/cli/lightning_cli_advanced_2.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/cli/lightning_cli_advanced_2.rst -------------------------------------------------------------------------------- /docs/source-pytorch/cli/lightning_cli_advanced_3.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/cli/lightning_cli_advanced_3.rst -------------------------------------------------------------------------------- /docs/source-pytorch/cli/lightning_cli_expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/cli/lightning_cli_expert.rst -------------------------------------------------------------------------------- /docs/source-pytorch/cli/lightning_cli_faq.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/cli/lightning_cli_faq.rst -------------------------------------------------------------------------------- /docs/source-pytorch/clouds/cluster.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/clouds/cluster.rst -------------------------------------------------------------------------------- /docs/source-pytorch/clouds/cluster_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/clouds/cluster_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/clouds/cluster_expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/clouds/cluster_expert.rst -------------------------------------------------------------------------------- /docs/source-pytorch/clouds/lightning_ai.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/clouds/lightning_ai.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/checkpointing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/checkpointing.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/checkpointing_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/checkpointing_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/checkpointing_expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/checkpointing_expert.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/child_modules.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/child_modules.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/console_logs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/console_logs.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/early_stopping.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/early_stopping.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/evaluation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/evaluation.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/evaluation_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/evaluation_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/hooks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/hooks.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/hyperparameters.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/hyperparameters.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/index.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/lightning_module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/lightning_module.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/notebooks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/notebooks.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/optimization.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/optimization.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/precision.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/precision_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/precision_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/precision_expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/precision_expert.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/progress_bar.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/progress_bar.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/remote_fs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/remote_fs.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/tbptt.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/tbptt.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common/trainer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common/trainer.rst -------------------------------------------------------------------------------- /docs/source-pytorch/common_usecases.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/common_usecases.rst -------------------------------------------------------------------------------- /docs/source-pytorch/community/governance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/community/governance.rst -------------------------------------------------------------------------------- /docs/source-pytorch/community/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/community/index.rst -------------------------------------------------------------------------------- /docs/source-pytorch/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/conf.py -------------------------------------------------------------------------------- /docs/source-pytorch/data/access.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/data/access.rst -------------------------------------------------------------------------------- /docs/source-pytorch/data/alternatives.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/data/alternatives.rst -------------------------------------------------------------------------------- /docs/source-pytorch/data/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/data/data.rst -------------------------------------------------------------------------------- /docs/source-pytorch/data/datamodule.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/data/datamodule.rst -------------------------------------------------------------------------------- /docs/source-pytorch/data/iterables.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/data/iterables.rst -------------------------------------------------------------------------------- /docs/source-pytorch/debug/debugging.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/debug/debugging.rst -------------------------------------------------------------------------------- /docs/source-pytorch/debug/debugging_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/debug/debugging_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/debug/debugging_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/debug/debugging_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/deploy/production.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/deploy/production.rst -------------------------------------------------------------------------------- /docs/source-pytorch/deploy/production_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/deploy/production_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/deploy/production_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/deploy/production_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/expertise_levels.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/expertise_levels.rst -------------------------------------------------------------------------------- /docs/source-pytorch/extensions/accelerator.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/extensions/accelerator.rst -------------------------------------------------------------------------------- /docs/source-pytorch/extensions/callbacks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/extensions/callbacks.rst -------------------------------------------------------------------------------- /docs/source-pytorch/extensions/callbacks_state.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/extensions/callbacks_state.rst -------------------------------------------------------------------------------- /docs/source-pytorch/extensions/entry_points.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/extensions/entry_points.rst -------------------------------------------------------------------------------- /docs/source-pytorch/extensions/logging.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/extensions/logging.rst -------------------------------------------------------------------------------- /docs/source-pytorch/extensions/plugins.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/extensions/plugins.rst -------------------------------------------------------------------------------- /docs/source-pytorch/extensions/strategy.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/extensions/strategy.rst -------------------------------------------------------------------------------- /docs/source-pytorch/glossary/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/glossary/index.rst -------------------------------------------------------------------------------- /docs/source-pytorch/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/index.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/advanced_level_15.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/advanced_level_15.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/advanced_level_16.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/advanced_level_16.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/advanced_level_17.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/advanced_level_17.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/advanced_level_18.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/advanced_level_18.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/advanced_level_20.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/advanced_level_20.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/advanced_level_21.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/advanced_level_21.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/basic_level_2.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/basic_level_2.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/basic_level_5.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/basic_level_5.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/core_level_3.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/core_level_3.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/core_level_6.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/core_level_6.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/core_skills.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/core_skills.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/expert.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/expert_level_22.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/expert_level_22.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/expert_level_23.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/expert_level_23.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/expert_level_24.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/expert_level_24.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/intermediate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/intermediate.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/intermediate_level_7.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/intermediate_level_7.rst -------------------------------------------------------------------------------- /docs/source-pytorch/levels/intermediate_level_9.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/levels/intermediate_level_9.rst -------------------------------------------------------------------------------- /docs/source-pytorch/links.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/links.rst -------------------------------------------------------------------------------- /docs/source-pytorch/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/make.bat -------------------------------------------------------------------------------- /docs/source-pytorch/model/build_model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/model/build_model.rst -------------------------------------------------------------------------------- /docs/source-pytorch/model/build_model_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/model/build_model_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/model/manual_optimization.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/model/manual_optimization.rst -------------------------------------------------------------------------------- /docs/source-pytorch/model/own_your_loop.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/model/own_your_loop.rst -------------------------------------------------------------------------------- /docs/source-pytorch/model/train_model_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/model/train_model_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/notebooks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/notebooks.rst -------------------------------------------------------------------------------- /docs/source-pytorch/past_versions.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/past_versions.rst -------------------------------------------------------------------------------- /docs/source-pytorch/starter/converting.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/starter/converting.rst -------------------------------------------------------------------------------- /docs/source-pytorch/starter/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/starter/installation.rst -------------------------------------------------------------------------------- /docs/source-pytorch/starter/introduction.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/starter/introduction.rst -------------------------------------------------------------------------------- /docs/source-pytorch/starter/style_guide.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/starter/style_guide.rst -------------------------------------------------------------------------------- /docs/source-pytorch/tuning/profiler.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/tuning/profiler.rst -------------------------------------------------------------------------------- /docs/source-pytorch/tuning/profiler_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/tuning/profiler_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/tuning/profiler_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/tuning/profiler_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/tuning/profiler_expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/tuning/profiler_expert.rst -------------------------------------------------------------------------------- /docs/source-pytorch/tutorials.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/tutorials.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/from_1_4.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/from_1_4.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/from_1_5.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/from_1_5.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/from_1_6.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/from_1_6.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/from_1_7.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/from_1_7.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/from_1_8.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/from_1_8.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/from_1_9.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/from_1_9.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/from_2_0.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/from_2_0.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/migration_guide.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/migration_guide.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/sections/1_4_devel.rst: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/sections/1_5_devel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/sections/1_5_devel.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/sections/1_6_devel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/sections/1_6_devel.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/sections/1_7_devel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/sections/1_7_devel.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/sections/1_8_devel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/sections/1_8_devel.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/sections/1_9_devel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/sections/1_9_devel.rst -------------------------------------------------------------------------------- /docs/source-pytorch/upgrade/sections/2_0_devel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/upgrade/sections/2_0_devel.rst -------------------------------------------------------------------------------- /docs/source-pytorch/versioning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/versioning.rst -------------------------------------------------------------------------------- /docs/source-pytorch/visualize/loggers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/visualize/loggers.rst -------------------------------------------------------------------------------- /docs/source-pytorch/visualize/logging_advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/visualize/logging_advanced.rst -------------------------------------------------------------------------------- /docs/source-pytorch/visualize/logging_basic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/visualize/logging_basic.rst -------------------------------------------------------------------------------- /docs/source-pytorch/visualize/logging_expert.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/docs/source-pytorch/visualize/logging_expert.rst -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/fabric/build_your_own_trainer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/build_your_own_trainer/README.md -------------------------------------------------------------------------------- /examples/fabric/build_your_own_trainer/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/build_your_own_trainer/run.py -------------------------------------------------------------------------------- /examples/fabric/build_your_own_trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/build_your_own_trainer/trainer.py -------------------------------------------------------------------------------- /examples/fabric/dcgan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/dcgan/README.md -------------------------------------------------------------------------------- /examples/fabric/dcgan/train_fabric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/dcgan/train_fabric.py -------------------------------------------------------------------------------- /examples/fabric/dcgan/train_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/dcgan/train_torch.py -------------------------------------------------------------------------------- /examples/fabric/fp8_distributed_transformer/requirements.txt: -------------------------------------------------------------------------------- 1 | torchao>=0.7.0 2 | -------------------------------------------------------------------------------- /examples/fabric/image_classifier/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/image_classifier/README.md -------------------------------------------------------------------------------- /examples/fabric/image_classifier/train_fabric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/image_classifier/train_fabric.py -------------------------------------------------------------------------------- /examples/fabric/image_classifier/train_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/image_classifier/train_torch.py -------------------------------------------------------------------------------- /examples/fabric/kfold_cv/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/kfold_cv/README.md -------------------------------------------------------------------------------- /examples/fabric/kfold_cv/train_fabric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/kfold_cv/train_fabric.py -------------------------------------------------------------------------------- /examples/fabric/language_model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/language_model/README.md -------------------------------------------------------------------------------- /examples/fabric/language_model/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/language_model/train.py -------------------------------------------------------------------------------- /examples/fabric/meta_learning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/meta_learning/README.md -------------------------------------------------------------------------------- /examples/fabric/meta_learning/train_fabric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/meta_learning/train_fabric.py -------------------------------------------------------------------------------- /examples/fabric/meta_learning/train_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/meta_learning/train_torch.py -------------------------------------------------------------------------------- /examples/fabric/reinforcement_learning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/reinforcement_learning/README.md -------------------------------------------------------------------------------- /examples/fabric/reinforcement_learning/requirements.txt: -------------------------------------------------------------------------------- 1 | gymnasium[box2d]>=0.27.1 2 | moviepy 3 | lightning>=1.9.0 4 | torchmetrics 5 | tensorboard 6 | -------------------------------------------------------------------------------- /examples/fabric/reinforcement_learning/rl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/fabric/reinforcement_learning/rl/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/reinforcement_learning/rl/agent.py -------------------------------------------------------------------------------- /examples/fabric/reinforcement_learning/rl/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/reinforcement_learning/rl/loss.py -------------------------------------------------------------------------------- /examples/fabric/reinforcement_learning/rl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/reinforcement_learning/rl/utils.py -------------------------------------------------------------------------------- /examples/fabric/tensor_parallel/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/tensor_parallel/README.md -------------------------------------------------------------------------------- /examples/fabric/tensor_parallel/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/tensor_parallel/data.py -------------------------------------------------------------------------------- /examples/fabric/tensor_parallel/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/tensor_parallel/model.py -------------------------------------------------------------------------------- /examples/fabric/tensor_parallel/parallelism.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/tensor_parallel/parallelism.py -------------------------------------------------------------------------------- /examples/fabric/tensor_parallel/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/fabric/tensor_parallel/train.py -------------------------------------------------------------------------------- /examples/pytorch/basics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/basics/README.md -------------------------------------------------------------------------------- /examples/pytorch/basics/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/basics/autoencoder.py -------------------------------------------------------------------------------- /examples/pytorch/basics/profiler_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/basics/profiler_example.py -------------------------------------------------------------------------------- /examples/pytorch/basics/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/basics/transformer.py -------------------------------------------------------------------------------- /examples/pytorch/bug_report/bug_report_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/bug_report/bug_report_model.ipynb -------------------------------------------------------------------------------- /examples/pytorch/bug_report/bug_report_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/bug_report/bug_report_model.py -------------------------------------------------------------------------------- /examples/pytorch/domain_templates/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/domain_templates/imagenet.py -------------------------------------------------------------------------------- /examples/pytorch/fp8_distributed_transformer/requirements.txt: -------------------------------------------------------------------------------- 1 | torchao>=0.7.0 2 | -------------------------------------------------------------------------------- /examples/pytorch/servable_module/production.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/servable_module/production.py -------------------------------------------------------------------------------- /examples/pytorch/tensor_parallel/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/tensor_parallel/README.md -------------------------------------------------------------------------------- /examples/pytorch/tensor_parallel/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/tensor_parallel/data.py -------------------------------------------------------------------------------- /examples/pytorch/tensor_parallel/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/tensor_parallel/model.py -------------------------------------------------------------------------------- /examples/pytorch/tensor_parallel/parallelism.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/tensor_parallel/parallelism.py -------------------------------------------------------------------------------- /examples/pytorch/tensor_parallel/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/pytorch/tensor_parallel/train.py -------------------------------------------------------------------------------- /examples/run_fabric_examples.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/run_fabric_examples.sh -------------------------------------------------------------------------------- /examples/run_pl_examples.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/examples/run_pl_examples.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/README.md -------------------------------------------------------------------------------- /requirements/ci.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/ci.txt -------------------------------------------------------------------------------- /requirements/collect_env_details.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/collect_env_details.py -------------------------------------------------------------------------------- /requirements/data/data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/data/data.txt -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/doctests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/doctests.txt -------------------------------------------------------------------------------- /requirements/fabric/base.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/fabric/base.txt -------------------------------------------------------------------------------- /requirements/fabric/docs.txt: -------------------------------------------------------------------------------- 1 | -r ../docs.txt 2 | 3 | tensorboard 4 | -------------------------------------------------------------------------------- /requirements/fabric/examples.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/fabric/examples.txt -------------------------------------------------------------------------------- /requirements/fabric/extra.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/fabric/extra.txt -------------------------------------------------------------------------------- /requirements/fabric/strategies.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/fabric/strategies.txt -------------------------------------------------------------------------------- /requirements/fabric/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/fabric/test.txt -------------------------------------------------------------------------------- /requirements/pytorch/base.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/base.txt -------------------------------------------------------------------------------- /requirements/pytorch/check-avail-extras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/check-avail-extras.py -------------------------------------------------------------------------------- /requirements/pytorch/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/docs.txt -------------------------------------------------------------------------------- /requirements/pytorch/examples.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/examples.txt -------------------------------------------------------------------------------- /requirements/pytorch/extra.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/extra.txt -------------------------------------------------------------------------------- /requirements/pytorch/loggers.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/loggers.info -------------------------------------------------------------------------------- /requirements/pytorch/strategies.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/strategies.txt -------------------------------------------------------------------------------- /requirements/pytorch/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/pytorch/test.txt -------------------------------------------------------------------------------- /requirements/typing.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/requirements/typing.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/setup.py -------------------------------------------------------------------------------- /src/lightning/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/MANIFEST.in -------------------------------------------------------------------------------- /src/lightning/__about__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/__about__.py -------------------------------------------------------------------------------- /src/lightning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/__init__.py -------------------------------------------------------------------------------- /src/lightning/__setup__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/__setup__.py -------------------------------------------------------------------------------- /src/lightning/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/__version__.py -------------------------------------------------------------------------------- /src/lightning/data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/data/README.md -------------------------------------------------------------------------------- /src/lightning/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/data/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/CHANGELOG.md -------------------------------------------------------------------------------- /src/lightning/fabric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/_graveyard/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/_graveyard/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/_graveyard/tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/_graveyard/tpu.py -------------------------------------------------------------------------------- /src/lightning/fabric/accelerators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/accelerators/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/accelerators/accelerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/accelerators/accelerator.py -------------------------------------------------------------------------------- /src/lightning/fabric/accelerators/cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/accelerators/cpu.py -------------------------------------------------------------------------------- /src/lightning/fabric/accelerators/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/accelerators/cuda.py -------------------------------------------------------------------------------- /src/lightning/fabric/accelerators/mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/accelerators/mps.py -------------------------------------------------------------------------------- /src/lightning/fabric/accelerators/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/accelerators/registry.py -------------------------------------------------------------------------------- /src/lightning/fabric/accelerators/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/accelerators/xla.py -------------------------------------------------------------------------------- /src/lightning/fabric/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/cli.py -------------------------------------------------------------------------------- /src/lightning/fabric/connector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/connector.py -------------------------------------------------------------------------------- /src/lightning/fabric/fabric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/fabric.py -------------------------------------------------------------------------------- /src/lightning/fabric/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/loggers/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/loggers/csv_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/loggers/csv_logs.py -------------------------------------------------------------------------------- /src/lightning/fabric/loggers/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/loggers/logger.py -------------------------------------------------------------------------------- /src/lightning/fabric/loggers/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/loggers/tensorboard.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/environments/lsf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/environments/lsf.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/environments/mpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/environments/mpi.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/environments/slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/environments/slurm.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/environments/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/environments/xla.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/io/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/io/checkpoint_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/io/checkpoint_io.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/io/torch_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/io/torch_io.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/io/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/io/xla.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/amp.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/deepspeed.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/double.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/double.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/fsdp.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/half.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/precision.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/utils.py -------------------------------------------------------------------------------- /src/lightning/fabric/plugins/precision/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/plugins/precision/xla.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/ddp.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/deepspeed.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/dp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/dp.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/fsdp.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/launchers/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/launchers/xla.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/model_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/model_parallel.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/parallel.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/registry.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/single_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/single_device.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/single_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/single_xla.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/strategy.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/xla.py -------------------------------------------------------------------------------- /src/lightning/fabric/strategies/xla_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/strategies/xla_fsdp.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/apply_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/apply_func.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/cloud_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/cloud_io.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/data.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/device_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/device_parser.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/distributed.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/enums.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/exceptions.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/imports.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/init.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/load.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/logger.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/optimizer.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/rank_zero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/rank_zero.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/registry.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/seed.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/spike.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/spike.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/testing/__init__.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/testing/_runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/testing/_runif.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/throughput.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/types.py -------------------------------------------------------------------------------- /src/lightning/fabric/utilities/warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/utilities/warnings.py -------------------------------------------------------------------------------- /src/lightning/fabric/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/fabric/wrappers.py -------------------------------------------------------------------------------- /src/lightning/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/lightning/pytorch/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/CHANGELOG.md -------------------------------------------------------------------------------- /src/lightning/pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/_graveyard/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/_graveyard/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/_graveyard/_torchmetrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/_graveyard/_torchmetrics.py -------------------------------------------------------------------------------- /src/lightning/pytorch/_graveyard/hpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/_graveyard/hpu.py -------------------------------------------------------------------------------- /src/lightning/pytorch/_graveyard/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/_graveyard/precision.py -------------------------------------------------------------------------------- /src/lightning/pytorch/_graveyard/tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/_graveyard/tpu.py -------------------------------------------------------------------------------- /src/lightning/pytorch/accelerators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/accelerators/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/accelerators/accelerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/accelerators/accelerator.py -------------------------------------------------------------------------------- /src/lightning/pytorch/accelerators/cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/accelerators/cpu.py -------------------------------------------------------------------------------- /src/lightning/pytorch/accelerators/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/accelerators/cuda.py -------------------------------------------------------------------------------- /src/lightning/pytorch/accelerators/mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/accelerators/mps.py -------------------------------------------------------------------------------- /src/lightning/pytorch/accelerators/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/accelerators/xla.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/callback.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/checkpoint.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/early_stopping.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/finetuning.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/lambda_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/lambda_function.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/lr_finder.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/lr_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/lr_monitor.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/model_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/model_checkpoint.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/model_summary.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/pruning.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/spike.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/spike.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/timer.py -------------------------------------------------------------------------------- /src/lightning/pytorch/callbacks/weight_averaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/callbacks/weight_averaging.py -------------------------------------------------------------------------------- /src/lightning/pytorch/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/cli.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/datamodule.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/hooks.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/mixins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/mixins/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/mixins/hparams_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/mixins/hparams_mixin.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/module.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/optimizer.py -------------------------------------------------------------------------------- /src/lightning/pytorch/core/saving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/core/saving.py -------------------------------------------------------------------------------- /src/lightning/pytorch/demos/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/demos/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/demos/boring_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/demos/boring_classes.py -------------------------------------------------------------------------------- /src/lightning/pytorch/demos/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/demos/lstm.py -------------------------------------------------------------------------------- /src/lightning/pytorch/demos/mnist_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/demos/mnist_datamodule.py -------------------------------------------------------------------------------- /src/lightning/pytorch/demos/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/demos/transformer.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/comet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/comet.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/csv_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/csv_logs.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/logger.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/mlflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/mlflow.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/neptune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/neptune.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/tensorboard.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/utilities.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loggers/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loggers/wandb.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/evaluation_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/evaluation_loop.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/fetchers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/fetchers.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/fit_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/fit_loop.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/loop.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/optimization/closure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/optimization/closure.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/optimization/manual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/optimization/manual.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/prediction_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/prediction_loop.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/progress.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/training_epoch_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/training_epoch_loop.py -------------------------------------------------------------------------------- /src/lightning/pytorch/loops/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/loops/utilities.py -------------------------------------------------------------------------------- /src/lightning/pytorch/overrides/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/lightning/pytorch/overrides/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/overrides/distributed.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/io/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/io/async_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/io/async_plugin.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/io/torch_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/io/torch_plugin.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/io/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/io/wrapper.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/io/xla_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/io/xla_plugin.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/layer_sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/layer_sync.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/precision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/precision/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/precision/amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/precision/amp.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/precision/double.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/precision/double.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/precision/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/precision/fsdp.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/precision/half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/precision/half.py -------------------------------------------------------------------------------- /src/lightning/pytorch/plugins/precision/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/plugins/precision/xla.py -------------------------------------------------------------------------------- /src/lightning/pytorch/profilers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/profilers/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/profilers/advanced.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/profilers/advanced.py -------------------------------------------------------------------------------- /src/lightning/pytorch/profilers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/profilers/base.py -------------------------------------------------------------------------------- /src/lightning/pytorch/profilers/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/profilers/profiler.py -------------------------------------------------------------------------------- /src/lightning/pytorch/profilers/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/profilers/pytorch.py -------------------------------------------------------------------------------- /src/lightning/pytorch/profilers/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/profilers/simple.py -------------------------------------------------------------------------------- /src/lightning/pytorch/profilers/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/profilers/xla.py -------------------------------------------------------------------------------- /src/lightning/pytorch/serve/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/serve/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/serve/servable_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/serve/servable_module.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/ddp.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/deepspeed.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/fsdp.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/launchers/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/launchers/xla.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/model_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/model_parallel.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/parallel.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/single_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/single_device.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/single_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/single_xla.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/strategy.py -------------------------------------------------------------------------------- /src/lightning/pytorch/strategies/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/strategies/xla.py -------------------------------------------------------------------------------- /src/lightning/pytorch/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/trainer/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/trainer/call.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/trainer/call.py -------------------------------------------------------------------------------- /src/lightning/pytorch/trainer/connectors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/lightning/pytorch/trainer/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/trainer/setup.py -------------------------------------------------------------------------------- /src/lightning/pytorch/trainer/states.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/trainer/states.py -------------------------------------------------------------------------------- /src/lightning/pytorch/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/trainer/trainer.py -------------------------------------------------------------------------------- /src/lightning/pytorch/tuner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/tuner/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/tuner/batch_size_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/tuner/batch_size_scaling.py -------------------------------------------------------------------------------- /src/lightning/pytorch/tuner/lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/tuner/lr_finder.py -------------------------------------------------------------------------------- /src/lightning/pytorch/tuner/tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/tuner/tuning.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/_pytree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/_pytree.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/argparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/argparse.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/combined_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/combined_loader.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/compile.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/data.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/deepspeed.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/enums.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/exceptions.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/grads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/grads.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/imports.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/memory.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/migration/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/migration/utils.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/model_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/model_helpers.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/model_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/model_registry.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/parameter_tying.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/parameter_tying.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/parsing.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/rank_zero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/rank_zero.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/seed.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/signature_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/signature_utils.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/testing/__init__.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/testing/_runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/testing/_runif.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/types.py -------------------------------------------------------------------------------- /src/lightning/pytorch/utilities/warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning/pytorch/utilities/warnings.py -------------------------------------------------------------------------------- /src/lightning_fabric/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning_fabric/MANIFEST.in -------------------------------------------------------------------------------- /src/lightning_fabric/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning_fabric/README.md -------------------------------------------------------------------------------- /src/lightning_fabric/__about__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning_fabric/__about__.py -------------------------------------------------------------------------------- /src/lightning_fabric/__setup__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning_fabric/__setup__.py -------------------------------------------------------------------------------- /src/lightning_fabric/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/lightning_fabric/__version__.py -------------------------------------------------------------------------------- /src/lightning_fabric/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/pytorch_lightning/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/pytorch_lightning/MANIFEST.in -------------------------------------------------------------------------------- /src/pytorch_lightning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/pytorch_lightning/README.md -------------------------------------------------------------------------------- /src/pytorch_lightning/__about__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/pytorch_lightning/__about__.py -------------------------------------------------------------------------------- /src/pytorch_lightning/__setup__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/pytorch_lightning/__setup__.py -------------------------------------------------------------------------------- /src/pytorch_lightning/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/src/pytorch_lightning/__version__.py -------------------------------------------------------------------------------- /src/pytorch_lightning/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/version.info: -------------------------------------------------------------------------------- 1 | 2.6.0 2 | -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/legacy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/legacy/README.md -------------------------------------------------------------------------------- /tests/legacy/back-compatible-versions.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/legacy/back-compatible-versions.txt -------------------------------------------------------------------------------- /tests/legacy/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/legacy/generate_checkpoints.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/legacy/generate_checkpoints.sh -------------------------------------------------------------------------------- /tests/legacy/simple_classif_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/legacy/simple_classif_training.py -------------------------------------------------------------------------------- /tests/parity_fabric/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/parity_fabric/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_fabric/conftest.py -------------------------------------------------------------------------------- /tests/parity_fabric/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_fabric/models.py -------------------------------------------------------------------------------- /tests/parity_fabric/run_standalone_tasks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_fabric/run_standalone_tasks.sh -------------------------------------------------------------------------------- /tests/parity_fabric/test_parity_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_fabric/test_parity_ddp.py -------------------------------------------------------------------------------- /tests/parity_fabric/test_parity_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_fabric/test_parity_simple.py -------------------------------------------------------------------------------- /tests/parity_fabric/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_fabric/utils.py -------------------------------------------------------------------------------- /tests/parity_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_pytorch/__init__.py -------------------------------------------------------------------------------- /tests/parity_pytorch/generate_comparison.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_pytorch/generate_comparison.py -------------------------------------------------------------------------------- /tests/parity_pytorch/measure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_pytorch/measure.py -------------------------------------------------------------------------------- /tests/parity_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_pytorch/models.py -------------------------------------------------------------------------------- /tests/parity_pytorch/test_basic_parity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_pytorch/test_basic_parity.py -------------------------------------------------------------------------------- /tests/parity_pytorch/test_sync_batchnorm_parity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/parity_pytorch/test_sync_batchnorm_parity.py -------------------------------------------------------------------------------- /tests/tests_fabric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/__init__.py -------------------------------------------------------------------------------- /tests/tests_fabric/accelerators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/accelerators/test_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/accelerators/test_cpu.py -------------------------------------------------------------------------------- /tests/tests_fabric/accelerators/test_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/accelerators/test_cuda.py -------------------------------------------------------------------------------- /tests/tests_fabric/accelerators/test_mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/accelerators/test_mps.py -------------------------------------------------------------------------------- /tests/tests_fabric/accelerators/test_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/accelerators/test_registry.py -------------------------------------------------------------------------------- /tests/tests_fabric/accelerators/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/accelerators/test_xla.py -------------------------------------------------------------------------------- /tests/tests_fabric/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/conftest.py -------------------------------------------------------------------------------- /tests/tests_fabric/graveyard/test_tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/graveyard/test_tpu.py -------------------------------------------------------------------------------- /tests/tests_fabric/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/helpers/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/helpers/dataloaders.py -------------------------------------------------------------------------------- /tests/tests_fabric/helpers/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/helpers/datasets.py -------------------------------------------------------------------------------- /tests/tests_fabric/helpers/runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/helpers/runif.py -------------------------------------------------------------------------------- /tests/tests_fabric/loggers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/loggers/test_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/loggers/test_csv.py -------------------------------------------------------------------------------- /tests/tests_fabric/loggers/test_tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/loggers/test_tensorboard.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/collectives/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/environments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/environments/test_lsf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/environments/test_lsf.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/environments/test_mpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/environments/test_mpi.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/environments/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/environments/test_xla.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/test_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/precision/test_all.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/test_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/precision/test_amp.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/test_double.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/precision/test_double.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/test_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/precision/test_fsdp.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/test_half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/precision/test_half.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/precision/test_utils.py -------------------------------------------------------------------------------- /tests/tests_fabric/plugins/precision/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/plugins/precision/test_xla.py -------------------------------------------------------------------------------- /tests/tests_fabric/run_tpu_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/run_tpu_tests.sh -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/launchers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_ddp.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_deepspeed.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_dp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_dp.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_fsdp.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_registry.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_single_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_single_device.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_strategy.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_xla.py -------------------------------------------------------------------------------- /tests/tests_fabric/strategies/test_xla_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/strategies/test_xla_fsdp.py -------------------------------------------------------------------------------- /tests/tests_fabric/test_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/test_cli.py -------------------------------------------------------------------------------- /tests/tests_fabric/test_connector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/test_connector.py -------------------------------------------------------------------------------- /tests/tests_fabric/test_fabric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/test_fabric.py -------------------------------------------------------------------------------- /tests/tests_fabric/test_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/test_wrappers.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_apply_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_apply_func.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_cloud_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_cloud_io.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_data.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_device_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_device_parser.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_distributed.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_imports.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_init.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_load.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_logger.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_optimizer.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_rank_zero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_rank_zero.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_registry.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_runif.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_seed.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_spike.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_spike.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_throughput.py -------------------------------------------------------------------------------- /tests/tests_fabric/utilities/test_warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_fabric/utilities/test_warnings.py -------------------------------------------------------------------------------- /tests/tests_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/__init__.py -------------------------------------------------------------------------------- /tests/tests_pytorch/accelerators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/accelerators/test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/accelerators/test_common.py -------------------------------------------------------------------------------- /tests/tests_pytorch/accelerators/test_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/accelerators/test_cpu.py -------------------------------------------------------------------------------- /tests/tests_pytorch/accelerators/test_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/accelerators/test_gpu.py -------------------------------------------------------------------------------- /tests/tests_pytorch/accelerators/test_mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/accelerators/test_mps.py -------------------------------------------------------------------------------- /tests/tests_pytorch/accelerators/test_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/accelerators/test_registry.py -------------------------------------------------------------------------------- /tests/tests_pytorch/accelerators/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/accelerators/test_xla.py -------------------------------------------------------------------------------- /tests/tests_pytorch/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/callbacks/test_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/callbacks/test_callbacks.py -------------------------------------------------------------------------------- /tests/tests_pytorch/callbacks/test_lr_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/callbacks/test_lr_monitor.py -------------------------------------------------------------------------------- /tests/tests_pytorch/callbacks/test_model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/callbacks/test_model_summary.py -------------------------------------------------------------------------------- /tests/tests_pytorch/callbacks/test_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/callbacks/test_pruning.py -------------------------------------------------------------------------------- /tests/tests_pytorch/callbacks/test_spike.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/callbacks/test_spike.py -------------------------------------------------------------------------------- /tests/tests_pytorch/callbacks/test_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/callbacks/test_timer.py -------------------------------------------------------------------------------- /tests/tests_pytorch/checkpointing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/conftest.py -------------------------------------------------------------------------------- /tests/tests_pytorch/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/core/test_datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/core/test_datamodules.py -------------------------------------------------------------------------------- /tests/tests_pytorch/core/test_lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/core/test_lightning_module.py -------------------------------------------------------------------------------- /tests/tests_pytorch/core/test_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/core/test_results.py -------------------------------------------------------------------------------- /tests/tests_pytorch/core/test_saving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/core/test_saving.py -------------------------------------------------------------------------------- /tests/tests_pytorch/demos/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/demos/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/demos/lstm.py -------------------------------------------------------------------------------- /tests/tests_pytorch/demos/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/demos/transformer.py -------------------------------------------------------------------------------- /tests/tests_pytorch/deprecated_api/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/deprecated_api/__init__.py -------------------------------------------------------------------------------- /tests/tests_pytorch/graveyard/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/graveyard/test_hpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/graveyard/test_hpu.py -------------------------------------------------------------------------------- /tests/tests_pytorch/graveyard/test_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/graveyard/test_precision.py -------------------------------------------------------------------------------- /tests/tests_pytorch/graveyard/test_tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/graveyard/test_tpu.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/__init__.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/advanced_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/advanced_models.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/dataloaders.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/datamodules.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/datasets.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/deterministic_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/deterministic_model.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/pipelines.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/runif.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/simple_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/simple_models.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/test_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/test_datasets.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/test_models.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/threading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/threading.py -------------------------------------------------------------------------------- /tests/tests_pytorch/helpers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/helpers/utils.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/conftest.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_all.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_comet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_comet.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_csv.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_logger.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_mlflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_mlflow.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_neptune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_neptune.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_tensorboard.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_utilities.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loggers/test_wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loggers/test_wandb.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/optimization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_all.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_evaluation_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_evaluation_loop.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_fetchers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_fetchers.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_flow_warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_flow_warnings.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_loop_state_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_loop_state_dict.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_loops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_loops.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_prediction_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_prediction_loop.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_progress.py -------------------------------------------------------------------------------- /tests/tests_pytorch/loops/test_training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/loops/test_training_loop.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/models/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/conf/config.yaml -------------------------------------------------------------------------------- /tests/tests_pytorch/models/conf/training/default.yaml: -------------------------------------------------------------------------------- 1 | # @package training 2 | log: "Something" 3 | -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_amp.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_cpu.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_ddp_fork_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_ddp_fork_amp.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_gpu.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_hooks.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_hparams.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_onnx.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_restore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_restore.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_torch_tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_torch_tensorrt.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_torchscript.py -------------------------------------------------------------------------------- /tests/tests_pytorch/models/test_tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/models/test_tpu.py -------------------------------------------------------------------------------- /tests/tests_pytorch/overrides/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/overrides/test_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/overrides/test_distributed.py -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/precision/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/precision/test_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/plugins/precision/test_all.py -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/precision/test_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/plugins/precision/test_amp.py -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/precision/test_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/plugins/precision/test_fsdp.py -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/precision/test_half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/plugins/precision/test_half.py -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/precision/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/plugins/precision/test_xla.py -------------------------------------------------------------------------------- /tests/tests_pytorch/plugins/test_amp_plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/plugins/test_amp_plugins.py -------------------------------------------------------------------------------- /tests/tests_pytorch/profilers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/profilers/test_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/profilers/test_profiler.py -------------------------------------------------------------------------------- /tests/tests_pytorch/profilers/test_xla_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/profilers/test_xla_profiler.py -------------------------------------------------------------------------------- /tests/tests_pytorch/run_standalone_tasks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/run_standalone_tasks.sh -------------------------------------------------------------------------------- /tests/tests_pytorch/run_tpu_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/run_tpu_tests.sh -------------------------------------------------------------------------------- /tests/tests_pytorch/serve/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/launchers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/strategies/test_common.py -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/test_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/strategies/test_ddp.py -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/test_deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/strategies/test_deepspeed.py -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/test_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/strategies/test_fsdp.py -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/test_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/strategies/test_registry.py -------------------------------------------------------------------------------- /tests/tests_pytorch/strategies/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/strategies/test_xla.py -------------------------------------------------------------------------------- /tests/tests_pytorch/test_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/test_cli.py -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/connectors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/dynamic_args/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/flags/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/flags/test_barebones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/trainer/flags/test_barebones.py -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/flags/test_env_vars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/trainer/flags/test_env_vars.py -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/logging_/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/optimization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/properties/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/test_dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/trainer/test_dataloaders.py -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/test_states.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/trainer/test_states.py -------------------------------------------------------------------------------- /tests/tests_pytorch/trainer/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/trainer/test_trainer.py -------------------------------------------------------------------------------- /tests/tests_pytorch/tuner/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/tuner/test_lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/tuner/test_lr_finder.py -------------------------------------------------------------------------------- /tests/tests_pytorch/tuner/test_scale_batch_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/tuner/test_scale_batch_size.py -------------------------------------------------------------------------------- /tests/tests_pytorch/tuner/test_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/tuner/test_tuning.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/migration/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_auto_restart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_auto_restart.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_compile.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_data.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_enums.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_exceptions.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_grads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_grads.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_imports.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_memory.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_model_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_model_helpers.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_model_summary.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_parsing.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_pytree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_pytree.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_seed.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_types.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/test_warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/test_warnings.py -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/testing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tests_pytorch/utilities/testing/test_runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/pytorch-lightning/HEAD/tests/tests_pytorch/utilities/testing/test_runif.py --------------------------------------------------------------------------------