├── .github ├── ISSUE_TEMPLATE │ ├── bug_template.md │ ├── doc_template.md │ └── feature_template.md ├── PULL_REQUEST_TEMPLATE │ └── pull_request_template.md └── workflows │ ├── doc-build.yml │ ├── main_cpp.yml │ ├── main_distributed.yaml │ └── main_python.yml ├── .gitignore ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── cpp ├── .clang-format ├── autograd │ ├── CMakeLists.txt │ ├── README.md │ └── autograd.cpp ├── custom-dataset │ ├── CMakeLists.txt │ ├── README.md │ ├── custom-dataset.cpp │ └── info.txt ├── dcgan │ ├── CMakeLists.txt │ ├── README.md │ ├── dcgan.cpp │ └── display_samples.py ├── distributed │ ├── CMakeLists.txt │ ├── README.md │ └── dist-mnist.cpp ├── mnist │ ├── CMakeLists.txt │ ├── README.md │ └── mnist.cpp ├── regression │ ├── CMakeLists.txt │ ├── README.md │ └── regression.cpp ├── tools │ ├── InstallingOpenCV.md │ └── download_mnist.py └── transfer-learning │ ├── CMakeLists.txt │ ├── README.md │ ├── classify.cpp │ ├── convert.py │ ├── main.cpp │ └── main.h ├── dcgan ├── README.md ├── main.py └── requirements.txt ├── distributed ├── FSDP │ ├── .gitignore │ ├── README.md │ ├── T5_training.py │ ├── configs │ │ ├── __init__.py │ │ ├── fsdp.py │ │ └── training.py │ ├── download_dataset.sh │ ├── model_checkpointing │ │ ├── __init__.py │ │ └── checkpoint_handler.py │ ├── policies │ │ ├── __init__.py │ │ ├── activation_checkpointing_functions.py │ │ ├── mixed_precision.py │ │ └── wrapping.py │ ├── requirements.txt │ ├── summarization_dataset.py │ └── utils │ │ ├── __init__.py │ │ ├── environment.py │ │ └── train_utils.py ├── FSDP2 │ ├── README.md │ ├── checkpoint.py │ ├── example.py │ ├── model.py │ ├── requirements.txt │ ├── run_example.sh │ └── utils.py ├── ddp-tutorial-series │ ├── README.md │ ├── datautils.py │ ├── multigpu.py │ ├── multigpu_torchrun.py │ ├── multinode.py │ ├── requirements.txt │ ├── single_gpu.py │ └── slurm │ │ ├── config.yaml.template │ │ ├── sbatch_run.sh │ │ └── setup_pcluster_slurm.md ├── ddp │ ├── README.md │ ├── example.py │ ├── requirements.txt │ └── run_example.sh ├── minGPT-ddp │ ├── README.md │ ├── mingpt │ │ ├── char_dataset.py │ │ ├── data │ │ │ └── input.txt │ │ ├── gpt2_train_cfg.yaml │ │ ├── main.py │ │ ├── model.py │ │ ├── slurm │ │ │ ├── config.yaml.template │ │ │ ├── sbatch_run.sh │ │ │ └── setup_pcluster_slurm.md │ │ └── trainer.py │ ├── requirements.txt │ └── run_example.sh ├── rpc │ ├── batch │ │ ├── README.md │ │ ├── parameter_server.py │ │ ├── reinforce.py │ │ └── requirements.txt │ ├── ddp_rpc │ │ ├── README.md │ │ ├── main.py │ │ └── requirements.txt │ ├── parameter_server │ │ ├── README.md │ │ ├── requirements.txt │ │ └── rpc_parameter_server.py │ ├── pipeline │ │ ├── README.md │ │ ├── main.py │ │ └── requirements.txt │ ├── rl │ │ ├── README.md │ │ ├── main.py │ │ └── requirements.txt │ └── rnn │ │ ├── README.md │ │ ├── main.py │ │ ├── requirements.txt │ │ └── rnn.py └── tensor_parallelism │ ├── README.md │ ├── fsdp_tp_example.py │ ├── llama2_model.py │ ├── log_utils.py │ ├── requirements.txt │ ├── run_example.sh │ ├── sequence_parallel_example.py │ └── tensor_parallel_example.py ├── docs ├── Makefile ├── make.bat ├── requirements.txt └── source │ ├── _static │ └── .gitkeep │ ├── conf.py │ └── index.rst ├── fast_neural_style ├── README.md ├── download_saved_models.py ├── images │ ├── content-images │ │ └── amber.jpg │ ├── output-images │ │ ├── amber-candy.jpg │ │ ├── amber-mosaic.jpg │ │ ├── amber-rain-princess.jpg │ │ └── amber-udnie.jpg │ └── style-images │ │ ├── candy.jpg │ │ ├── mosaic.jpg │ │ ├── rain-princess-cropped.jpg │ │ ├── rain-princess.jpg │ │ └── udnie.jpg ├── neural_style │ ├── __init__.py │ ├── neural_style.py │ ├── transformer_net.py │ ├── utils.py │ └── vgg.py └── requirements.txt ├── fx ├── README.md ├── custom_tracer.py ├── inline_function.py ├── invert.py ├── module_tracer.py ├── native_interpreter │ ├── CMakeLists.txt │ ├── README.md │ ├── interpreter.cpp │ └── use_interpreter.py ├── primitive_library.py ├── profiling_tracer.py ├── proxy_based_graph_creation.py ├── replace_op.py ├── requirements.txt ├── subgraph_rewriter_basic_use.py └── wrap_output_dynamically.py ├── gat ├── README.md ├── main.py └── requirements.txt ├── gcn ├── README.md ├── main.py └── requirements.txt ├── imagenet ├── README.md ├── extract_ILSVRC.sh ├── main.py └── requirements.txt ├── language_translation ├── README.md ├── main.py ├── requirements.txt └── src │ ├── data.py │ └── model.py ├── legacy └── snli │ ├── README.md │ ├── model.py │ ├── requirements.txt │ ├── train.py │ └── util.py ├── mnist ├── README.md ├── main.py └── requirements.txt ├── mnist_forward_forward ├── README.md ├── main.py └── requirements.txt ├── mnist_hogwild ├── README.md ├── main.py ├── requirements.txt └── train.py ├── mnist_rnn ├── README.md ├── main.py └── requirements.txt ├── regression ├── README.md ├── main.py └── requirements.txt ├── reinforcement_learning ├── README.md ├── actor_critic.py ├── reinforce.py └── requirements.txt ├── run_cpp_examples.sh ├── run_distributed_examples.sh ├── run_python_examples.sh ├── runtime.txt ├── siamese_network ├── README.md ├── main.py └── requirements.txt ├── super_resolution ├── README.md ├── data.py ├── dataset.py ├── main.py ├── model.py ├── requirements.txt └── super_resolve.py ├── time_sequence_prediction ├── README.md ├── generate_sine_wave.py ├── requirements.txt └── train.py ├── utils.sh ├── vae ├── README.md ├── main.py ├── requirements.txt └── results │ └── .gitignore └── word_language_model ├── README.md ├── data.py ├── data └── wikitext-2 │ ├── README │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── generate.py ├── main.py ├── model.py └── requirements.txt /.github/ISSUE_TEMPLATE/bug_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/ISSUE_TEMPLATE/bug_template.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/doc_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/ISSUE_TEMPLATE/doc_template.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/ISSUE_TEMPLATE/feature_template.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/PULL_REQUEST_TEMPLATE/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/doc-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/workflows/doc-build.yml -------------------------------------------------------------------------------- /.github/workflows/main_cpp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/workflows/main_cpp.yml -------------------------------------------------------------------------------- /.github/workflows/main_distributed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/workflows/main_distributed.yaml -------------------------------------------------------------------------------- /.github/workflows/main_python.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.github/workflows/main_python.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/.gitignore -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/README.md -------------------------------------------------------------------------------- /cpp/.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/.clang-format -------------------------------------------------------------------------------- /cpp/autograd/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/autograd/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/autograd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/autograd/README.md -------------------------------------------------------------------------------- /cpp/autograd/autograd.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/autograd/autograd.cpp -------------------------------------------------------------------------------- /cpp/custom-dataset/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/custom-dataset/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/custom-dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/custom-dataset/README.md -------------------------------------------------------------------------------- /cpp/custom-dataset/custom-dataset.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/custom-dataset/custom-dataset.cpp -------------------------------------------------------------------------------- /cpp/custom-dataset/info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/custom-dataset/info.txt -------------------------------------------------------------------------------- /cpp/dcgan/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/dcgan/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/dcgan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/dcgan/README.md -------------------------------------------------------------------------------- /cpp/dcgan/dcgan.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/dcgan/dcgan.cpp -------------------------------------------------------------------------------- /cpp/dcgan/display_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/dcgan/display_samples.py -------------------------------------------------------------------------------- /cpp/distributed/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/distributed/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/distributed/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/distributed/README.md -------------------------------------------------------------------------------- /cpp/distributed/dist-mnist.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/distributed/dist-mnist.cpp -------------------------------------------------------------------------------- /cpp/mnist/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/mnist/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/mnist/README.md -------------------------------------------------------------------------------- /cpp/mnist/mnist.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/mnist/mnist.cpp -------------------------------------------------------------------------------- /cpp/regression/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/regression/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/regression/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/regression/README.md -------------------------------------------------------------------------------- /cpp/regression/regression.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/regression/regression.cpp -------------------------------------------------------------------------------- /cpp/tools/InstallingOpenCV.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/tools/InstallingOpenCV.md -------------------------------------------------------------------------------- /cpp/tools/download_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/tools/download_mnist.py -------------------------------------------------------------------------------- /cpp/transfer-learning/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/transfer-learning/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/transfer-learning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/transfer-learning/README.md -------------------------------------------------------------------------------- /cpp/transfer-learning/classify.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/transfer-learning/classify.cpp -------------------------------------------------------------------------------- /cpp/transfer-learning/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/transfer-learning/convert.py -------------------------------------------------------------------------------- /cpp/transfer-learning/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/transfer-learning/main.cpp -------------------------------------------------------------------------------- /cpp/transfer-learning/main.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/cpp/transfer-learning/main.h -------------------------------------------------------------------------------- /dcgan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/dcgan/README.md -------------------------------------------------------------------------------- /dcgan/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/dcgan/main.py -------------------------------------------------------------------------------- /dcgan/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/dcgan/requirements.txt -------------------------------------------------------------------------------- /distributed/FSDP/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | *.pt 3 | *.csv -------------------------------------------------------------------------------- /distributed/FSDP/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/README.md -------------------------------------------------------------------------------- /distributed/FSDP/T5_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/T5_training.py -------------------------------------------------------------------------------- /distributed/FSDP/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/configs/__init__.py -------------------------------------------------------------------------------- /distributed/FSDP/configs/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/configs/fsdp.py -------------------------------------------------------------------------------- /distributed/FSDP/configs/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/configs/training.py -------------------------------------------------------------------------------- /distributed/FSDP/download_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/download_dataset.sh -------------------------------------------------------------------------------- /distributed/FSDP/model_checkpointing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/model_checkpointing/__init__.py -------------------------------------------------------------------------------- /distributed/FSDP/model_checkpointing/checkpoint_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/model_checkpointing/checkpoint_handler.py -------------------------------------------------------------------------------- /distributed/FSDP/policies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/policies/__init__.py -------------------------------------------------------------------------------- /distributed/FSDP/policies/activation_checkpointing_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/policies/activation_checkpointing_functions.py -------------------------------------------------------------------------------- /distributed/FSDP/policies/mixed_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/policies/mixed_precision.py -------------------------------------------------------------------------------- /distributed/FSDP/policies/wrapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/policies/wrapping.py -------------------------------------------------------------------------------- /distributed/FSDP/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/requirements.txt -------------------------------------------------------------------------------- /distributed/FSDP/summarization_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/summarization_dataset.py -------------------------------------------------------------------------------- /distributed/FSDP/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/utils/__init__.py -------------------------------------------------------------------------------- /distributed/FSDP/utils/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/utils/environment.py -------------------------------------------------------------------------------- /distributed/FSDP/utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP/utils/train_utils.py -------------------------------------------------------------------------------- /distributed/FSDP2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP2/README.md -------------------------------------------------------------------------------- /distributed/FSDP2/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP2/checkpoint.py -------------------------------------------------------------------------------- /distributed/FSDP2/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP2/example.py -------------------------------------------------------------------------------- /distributed/FSDP2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP2/model.py -------------------------------------------------------------------------------- /distributed/FSDP2/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=2.7 2 | numpy 3 | -------------------------------------------------------------------------------- /distributed/FSDP2/run_example.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP2/run_example.sh -------------------------------------------------------------------------------- /distributed/FSDP2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/FSDP2/utils.py -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/README.md -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/datautils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/datautils.py -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/multigpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/multigpu.py -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/multigpu_torchrun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/multigpu_torchrun.py -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/multinode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/multinode.py -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.11.0 -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/single_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/single_gpu.py -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/slurm/config.yaml.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/slurm/config.yaml.template -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/slurm/sbatch_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/slurm/sbatch_run.sh -------------------------------------------------------------------------------- /distributed/ddp-tutorial-series/slurm/setup_pcluster_slurm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp-tutorial-series/slurm/setup_pcluster_slurm.md -------------------------------------------------------------------------------- /distributed/ddp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp/README.md -------------------------------------------------------------------------------- /distributed/ddp/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp/example.py -------------------------------------------------------------------------------- /distributed/ddp/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=2.7 2 | -------------------------------------------------------------------------------- /distributed/ddp/run_example.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/ddp/run_example.sh -------------------------------------------------------------------------------- /distributed/minGPT-ddp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/README.md -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/char_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/char_dataset.py -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/data/input.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/data/input.txt -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/gpt2_train_cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/gpt2_train_cfg.yaml -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/main.py -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/model.py -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/slurm/config.yaml.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/slurm/config.yaml.template -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/slurm/sbatch_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/slurm/sbatch_run.sh -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/slurm/setup_pcluster_slurm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/slurm/setup_pcluster_slurm.md -------------------------------------------------------------------------------- /distributed/minGPT-ddp/mingpt/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/mingpt/trainer.py -------------------------------------------------------------------------------- /distributed/minGPT-ddp/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=2.7 2 | boto3 3 | hydra-core 4 | requests 5 | aiohttp 6 | -------------------------------------------------------------------------------- /distributed/minGPT-ddp/run_example.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/minGPT-ddp/run_example.sh -------------------------------------------------------------------------------- /distributed/rpc/batch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/batch/README.md -------------------------------------------------------------------------------- /distributed/rpc/batch/parameter_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/batch/parameter_server.py -------------------------------------------------------------------------------- /distributed/rpc/batch/reinforce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/batch/reinforce.py -------------------------------------------------------------------------------- /distributed/rpc/batch/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/batch/requirements.txt -------------------------------------------------------------------------------- /distributed/rpc/ddp_rpc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/ddp_rpc/README.md -------------------------------------------------------------------------------- /distributed/rpc/ddp_rpc/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/ddp_rpc/main.py -------------------------------------------------------------------------------- /distributed/rpc/ddp_rpc/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=2.7.0 2 | numpy 3 | -------------------------------------------------------------------------------- /distributed/rpc/parameter_server/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/parameter_server/README.md -------------------------------------------------------------------------------- /distributed/rpc/parameter_server/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=2.7.1 2 | numpy 3 | -------------------------------------------------------------------------------- /distributed/rpc/parameter_server/rpc_parameter_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/parameter_server/rpc_parameter_server.py -------------------------------------------------------------------------------- /distributed/rpc/pipeline/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/pipeline/README.md -------------------------------------------------------------------------------- /distributed/rpc/pipeline/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/pipeline/main.py -------------------------------------------------------------------------------- /distributed/rpc/pipeline/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/pipeline/requirements.txt -------------------------------------------------------------------------------- /distributed/rpc/rl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/rl/README.md -------------------------------------------------------------------------------- /distributed/rpc/rl/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/rl/main.py -------------------------------------------------------------------------------- /distributed/rpc/rl/requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | gymnasium 4 | -------------------------------------------------------------------------------- /distributed/rpc/rnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/rnn/README.md -------------------------------------------------------------------------------- /distributed/rpc/rnn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/rnn/main.py -------------------------------------------------------------------------------- /distributed/rpc/rnn/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=2.7.1 2 | numpy 3 | -------------------------------------------------------------------------------- /distributed/rpc/rnn/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/rpc/rnn/rnn.py -------------------------------------------------------------------------------- /distributed/tensor_parallelism/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/README.md -------------------------------------------------------------------------------- /distributed/tensor_parallelism/fsdp_tp_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/fsdp_tp_example.py -------------------------------------------------------------------------------- /distributed/tensor_parallelism/llama2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/llama2_model.py -------------------------------------------------------------------------------- /distributed/tensor_parallelism/log_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/log_utils.py -------------------------------------------------------------------------------- /distributed/tensor_parallelism/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/requirements.txt -------------------------------------------------------------------------------- /distributed/tensor_parallelism/run_example.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/run_example.sh -------------------------------------------------------------------------------- /distributed/tensor_parallelism/sequence_parallel_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/sequence_parallel_example.py -------------------------------------------------------------------------------- /distributed/tensor_parallelism/tensor_parallel_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/distributed/tensor_parallelism/tensor_parallel_example.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/_static/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /fast_neural_style/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/README.md -------------------------------------------------------------------------------- /fast_neural_style/download_saved_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/download_saved_models.py -------------------------------------------------------------------------------- /fast_neural_style/images/content-images/amber.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/content-images/amber.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/output-images/amber-candy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/output-images/amber-candy.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/output-images/amber-mosaic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/output-images/amber-mosaic.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/output-images/amber-rain-princess.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/output-images/amber-rain-princess.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/output-images/amber-udnie.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/output-images/amber-udnie.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/style-images/candy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/style-images/candy.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/style-images/mosaic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/style-images/mosaic.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/style-images/rain-princess-cropped.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/style-images/rain-princess-cropped.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/style-images/rain-princess.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/style-images/rain-princess.jpg -------------------------------------------------------------------------------- /fast_neural_style/images/style-images/udnie.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/images/style-images/udnie.jpg -------------------------------------------------------------------------------- /fast_neural_style/neural_style/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fast_neural_style/neural_style/neural_style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/neural_style/neural_style.py -------------------------------------------------------------------------------- /fast_neural_style/neural_style/transformer_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/neural_style/transformer_net.py -------------------------------------------------------------------------------- /fast_neural_style/neural_style/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/neural_style/utils.py -------------------------------------------------------------------------------- /fast_neural_style/neural_style/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/neural_style/vgg.py -------------------------------------------------------------------------------- /fast_neural_style/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fast_neural_style/requirements.txt -------------------------------------------------------------------------------- /fx/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/README.md -------------------------------------------------------------------------------- /fx/custom_tracer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/custom_tracer.py -------------------------------------------------------------------------------- /fx/inline_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/inline_function.py -------------------------------------------------------------------------------- /fx/invert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/invert.py -------------------------------------------------------------------------------- /fx/module_tracer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/module_tracer.py -------------------------------------------------------------------------------- /fx/native_interpreter/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/native_interpreter/CMakeLists.txt -------------------------------------------------------------------------------- /fx/native_interpreter/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/native_interpreter/README.md -------------------------------------------------------------------------------- /fx/native_interpreter/interpreter.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/native_interpreter/interpreter.cpp -------------------------------------------------------------------------------- /fx/native_interpreter/use_interpreter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/native_interpreter/use_interpreter.py -------------------------------------------------------------------------------- /fx/primitive_library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/primitive_library.py -------------------------------------------------------------------------------- /fx/profiling_tracer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/profiling_tracer.py -------------------------------------------------------------------------------- /fx/proxy_based_graph_creation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/proxy_based_graph_creation.py -------------------------------------------------------------------------------- /fx/replace_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/replace_op.py -------------------------------------------------------------------------------- /fx/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/requirements.txt -------------------------------------------------------------------------------- /fx/subgraph_rewriter_basic_use.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/subgraph_rewriter_basic_use.py -------------------------------------------------------------------------------- /fx/wrap_output_dynamically.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/fx/wrap_output_dynamically.py -------------------------------------------------------------------------------- /gat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/gat/README.md -------------------------------------------------------------------------------- /gat/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/gat/main.py -------------------------------------------------------------------------------- /gat/requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | requests 3 | numpy 4 | -------------------------------------------------------------------------------- /gcn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/gcn/README.md -------------------------------------------------------------------------------- /gcn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/gcn/main.py -------------------------------------------------------------------------------- /gcn/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/gcn/requirements.txt -------------------------------------------------------------------------------- /imagenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/imagenet/README.md -------------------------------------------------------------------------------- /imagenet/extract_ILSVRC.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/imagenet/extract_ILSVRC.sh -------------------------------------------------------------------------------- /imagenet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/imagenet/main.py -------------------------------------------------------------------------------- /imagenet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/imagenet/requirements.txt -------------------------------------------------------------------------------- /language_translation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/language_translation/README.md -------------------------------------------------------------------------------- /language_translation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/language_translation/main.py -------------------------------------------------------------------------------- /language_translation/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/language_translation/requirements.txt -------------------------------------------------------------------------------- /language_translation/src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/language_translation/src/data.py -------------------------------------------------------------------------------- /language_translation/src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/language_translation/src/model.py -------------------------------------------------------------------------------- /legacy/snli/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/legacy/snli/README.md -------------------------------------------------------------------------------- /legacy/snli/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/legacy/snli/model.py -------------------------------------------------------------------------------- /legacy/snli/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/legacy/snli/requirements.txt -------------------------------------------------------------------------------- /legacy/snli/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/legacy/snli/train.py -------------------------------------------------------------------------------- /legacy/snli/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/legacy/snli/util.py -------------------------------------------------------------------------------- /mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist/README.md -------------------------------------------------------------------------------- /mnist/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist/main.py -------------------------------------------------------------------------------- /mnist/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist/requirements.txt -------------------------------------------------------------------------------- /mnist_forward_forward/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_forward_forward/README.md -------------------------------------------------------------------------------- /mnist_forward_forward/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_forward_forward/main.py -------------------------------------------------------------------------------- /mnist_forward_forward/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_forward_forward/requirements.txt -------------------------------------------------------------------------------- /mnist_hogwild/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_hogwild/README.md -------------------------------------------------------------------------------- /mnist_hogwild/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_hogwild/main.py -------------------------------------------------------------------------------- /mnist_hogwild/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_hogwild/requirements.txt -------------------------------------------------------------------------------- /mnist_hogwild/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_hogwild/train.py -------------------------------------------------------------------------------- /mnist_rnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_rnn/README.md -------------------------------------------------------------------------------- /mnist_rnn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_rnn/main.py -------------------------------------------------------------------------------- /mnist_rnn/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/mnist_rnn/requirements.txt -------------------------------------------------------------------------------- /regression/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/regression/README.md -------------------------------------------------------------------------------- /regression/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/regression/main.py -------------------------------------------------------------------------------- /regression/requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | -------------------------------------------------------------------------------- /reinforcement_learning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/reinforcement_learning/README.md -------------------------------------------------------------------------------- /reinforcement_learning/actor_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/reinforcement_learning/actor_critic.py -------------------------------------------------------------------------------- /reinforcement_learning/reinforce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/reinforcement_learning/reinforce.py -------------------------------------------------------------------------------- /reinforcement_learning/requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | gymnasium[classic-control] 4 | -------------------------------------------------------------------------------- /run_cpp_examples.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/run_cpp_examples.sh -------------------------------------------------------------------------------- /run_distributed_examples.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/run_distributed_examples.sh -------------------------------------------------------------------------------- /run_python_examples.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/run_python_examples.sh -------------------------------------------------------------------------------- /runtime.txt: -------------------------------------------------------------------------------- 1 | 3.9 2 | -------------------------------------------------------------------------------- /siamese_network/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/siamese_network/README.md -------------------------------------------------------------------------------- /siamese_network/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/siamese_network/main.py -------------------------------------------------------------------------------- /siamese_network/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/siamese_network/requirements.txt -------------------------------------------------------------------------------- /super_resolution/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/super_resolution/README.md -------------------------------------------------------------------------------- /super_resolution/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/super_resolution/data.py -------------------------------------------------------------------------------- /super_resolution/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/super_resolution/dataset.py -------------------------------------------------------------------------------- /super_resolution/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/super_resolution/main.py -------------------------------------------------------------------------------- /super_resolution/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/super_resolution/model.py -------------------------------------------------------------------------------- /super_resolution/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/super_resolution/requirements.txt -------------------------------------------------------------------------------- /super_resolution/super_resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/super_resolution/super_resolve.py -------------------------------------------------------------------------------- /time_sequence_prediction/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/time_sequence_prediction/README.md -------------------------------------------------------------------------------- /time_sequence_prediction/generate_sine_wave.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/time_sequence_prediction/generate_sine_wave.py -------------------------------------------------------------------------------- /time_sequence_prediction/requirements.txt: -------------------------------------------------------------------------------- 1 | torch<2.6 2 | matplotlib 3 | -------------------------------------------------------------------------------- /time_sequence_prediction/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/time_sequence_prediction/train.py -------------------------------------------------------------------------------- /utils.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/utils.sh -------------------------------------------------------------------------------- /vae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/vae/README.md -------------------------------------------------------------------------------- /vae/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/vae/main.py -------------------------------------------------------------------------------- /vae/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/vae/requirements.txt -------------------------------------------------------------------------------- /vae/results/.gitignore: -------------------------------------------------------------------------------- 1 | *.png 2 | -------------------------------------------------------------------------------- /word_language_model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/README.md -------------------------------------------------------------------------------- /word_language_model/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/data.py -------------------------------------------------------------------------------- /word_language_model/data/wikitext-2/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/data/wikitext-2/README -------------------------------------------------------------------------------- /word_language_model/data/wikitext-2/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/data/wikitext-2/test.txt -------------------------------------------------------------------------------- /word_language_model/data/wikitext-2/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/data/wikitext-2/train.txt -------------------------------------------------------------------------------- /word_language_model/data/wikitext-2/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/data/wikitext-2/valid.txt -------------------------------------------------------------------------------- /word_language_model/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/generate.py -------------------------------------------------------------------------------- /word_language_model/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/main.py -------------------------------------------------------------------------------- /word_language_model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/examples/HEAD/word_language_model/model.py -------------------------------------------------------------------------------- /word_language_model/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=2.6 2 | --------------------------------------------------------------------------------