├── .coveragerc ├── .flake8 ├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.yml │ ├── documentation.yml │ ├── feature-request.yml │ └── help-support.yml ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── build_and_publish_docs.yaml │ ├── build_docs.yaml │ ├── nightly_build_cpu.yaml │ ├── pre_commit.yaml │ ├── release_build.yaml │ └── test.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── conftest.py ├── dev-requirements.txt ├── docs ├── .gitignore ├── Makefile ├── README.md ├── license_header.txt ├── requirements.txt └── source │ ├── _static │ ├── css │ │ └── torchtnt.css │ └── js │ │ ├── jquery.js │ │ └── torchtnt.js │ ├── assets │ └── TNTDiagram.png │ ├── checkpointing.rst │ ├── conf.py │ ├── distributed.rst │ ├── docutils.conf │ ├── examples.rst │ ├── ext │ └── fbcode.py │ ├── framework │ ├── auto_unit.rst │ ├── callbacks.rst │ ├── eval.rst │ ├── fit.rst │ ├── predict.rst │ ├── state.rst │ ├── train.rst │ └── unit.rst │ ├── index.rst │ ├── overview.rst │ ├── templates │ ├── class_template.rst │ └── layout.html │ └── utils │ └── utils.rst ├── examples ├── auto_unit_example.py ├── mingpt │ ├── char_dataset.py │ ├── data │ │ └── input.txt │ ├── main.py │ └── model.py ├── mnist │ ├── README.md │ ├── main.py │ └── requirements.txt ├── torchrec │ ├── README.md │ ├── main.py │ ├── requirements.txt │ └── tests │ │ └── torchrec_example_test.py └── train_unit_example.py ├── pyproject.toml ├── requirements.txt ├── setup.py ├── tests ├── framework │ ├── __init__.py │ ├── callbacks │ │ ├── test_base_checkpointer.py │ │ ├── test_checkpoint_utils.py │ │ ├── test_csv_writer.py │ │ ├── test_dcp_saver.py │ │ ├── test_dcp_saver_gpu.py │ │ ├── test_early_stopping.py │ │ ├── test_empty_cuda_cache.py │ │ ├── test_empty_dataloader_detector.py │ │ ├── test_garbage_collector.py │ │ ├── test_iteration_time_logger.py │ │ ├── test_lambda.py │ │ ├── test_learning_rate_monitor.py │ │ ├── test_memory_snapshot.py │ │ ├── test_module_summary.py │ │ ├── test_periodic_distributed_sync.py │ │ ├── test_progress_reporter.py │ │ ├── test_pytorch_profiler.py │ │ ├── test_slow_rank_detector.py │ │ ├── test_system_resources_monitor.py │ │ ├── test_tensorboard_parameter_monitor.py │ │ ├── test_tensorfloat32.py │ │ ├── test_throughput_logger.py │ │ ├── test_time_limit_interrupter.py │ │ ├── test_time_wait_for_batch_logger.py │ │ ├── test_torchsnapshot_saver.py │ │ ├── test_torchsnapshot_saver_gpu.py │ │ ├── test_tqdm_progress_bar.py │ │ └── test_train_progress_monitor.py │ ├── test_app_state_mixin.py │ ├── test_auto_unit.py │ ├── test_auto_unit_gpu.py │ ├── test_callback_handler.py │ ├── test_evaluate.py │ ├── test_fit.py │ ├── test_loop_utils.py │ ├── test_predict.py │ ├── test_state.py │ ├── test_train.py │ ├── test_unit.py │ ├── test_unit_utils.py │ ├── test_unit_utils_gpu.py │ └── test_utils.py └── utils │ ├── __init__.py │ ├── data │ ├── test_data_prefetcher.py │ ├── test_data_prefetcher_gpu.py │ ├── test_iterators.py │ ├── test_multi_dataloader.py │ └── test_profile_dataloader.py │ ├── loggers │ ├── __init__.py │ ├── test_anomaly_logger.py │ ├── test_csv.py │ ├── test_in_memory.py │ ├── test_json.py │ ├── test_stdout.py │ ├── test_tensorboard.py │ └── test_utils.py │ ├── test_anomaly_evaluation.py │ ├── test_checkpoint.py │ ├── test_checkpoint_gpu.py │ ├── test_device.py │ ├── test_device_gpu.py │ ├── test_device_mesh.py │ ├── test_distributed.py │ ├── test_distributed_gpu.py │ ├── test_early_stop_checker.py │ ├── test_early_stop_checker_gpu.py │ ├── test_env.py │ ├── test_flops.py │ ├── test_fsspec.py │ ├── test_memory.py │ ├── test_memory_snapshot_profiler.py │ ├── test_memory_snapshot_profiler_gpu.py │ ├── test_misc.py │ ├── test_module_summary.py │ ├── test_nan.py │ ├── test_oom.py │ ├── test_oom_gpu.py │ ├── test_optimizer.py │ ├── test_precision.py │ ├── test_prepare_module.py │ ├── test_prepare_module_gpu.py │ ├── test_progress.py │ ├── test_rank_zero_log.py │ ├── test_swa.py │ ├── test_timer.py │ ├── test_timer_gpu.py │ ├── test_tqdm.py │ └── test_version.py ├── torchtnt ├── __init__.py ├── framework │ ├── __init__.py │ ├── _callback_handler.py │ ├── _loop_utils.py │ ├── _test_utils.py │ ├── _unit_utils.py │ ├── auto_unit.py │ ├── callback.py │ ├── callbacks │ │ ├── __init__.py │ │ ├── _checkpoint_utils.py │ │ ├── base_checkpointer.py │ │ ├── base_csv_writer.py │ │ ├── checkpointer_types.py │ │ ├── dcp_saver.py │ │ ├── early_stopping.py │ │ ├── empty_cuda_cache.py │ │ ├── empty_dataloader_detector.py │ │ ├── garbage_collector.py │ │ ├── iteration_time_logger.py │ │ ├── lambda_callback.py │ │ ├── learning_rate_monitor.py │ │ ├── memory_snapshot.py │ │ ├── module_summary.py │ │ ├── periodic_distributed_sync.py │ │ ├── progress_reporter.py │ │ ├── pytorch_profiler.py │ │ ├── slow_rank_detector.py │ │ ├── system_resources_monitor.py │ │ ├── tensorboard_parameter_monitor.py │ │ ├── tensorfloat32.py │ │ ├── throughput_logger.py │ │ ├── time_limit_interrupter.py │ │ ├── time_wait_for_batch_logger.py │ │ ├── torch_compile.py │ │ ├── torchsnapshot_saver.py │ │ ├── tqdm_progress_bar.py │ │ └── train_progress_monitor.py │ ├── evaluate.py │ ├── fit.py │ ├── predict.py │ ├── state.py │ ├── train.py │ ├── unit.py │ └── utils.py ├── py.typed └── utils │ ├── __init__.py │ ├── anomaly_evaluation.py │ ├── checkpoint.py │ ├── data │ ├── __init__.py │ ├── data_prefetcher.py │ ├── iterators.py │ ├── multi_dataloader.py │ ├── profile_dataloader.py │ └── synthetic_data.py │ ├── device.py │ ├── device_mesh.py │ ├── distributed.py │ ├── early_stop_checker.py │ ├── env.py │ ├── event.py │ ├── event_handlers.py │ ├── flops.py │ ├── fsdp_utils.py │ ├── fsspec.py │ ├── loggers │ ├── __init__.py │ ├── anomaly_logger.py │ ├── csv.py │ ├── file.py │ ├── in_memory.py │ ├── json.py │ ├── logger.py │ ├── stdout.py │ ├── tensorboard.py │ └── utils.py │ ├── lr_scheduler.py │ ├── memory.py │ ├── memory_snapshot_profiler.py │ ├── misc.py │ ├── module_summary.py │ ├── nan.py │ ├── oom.py │ ├── optimizer.py │ ├── precision.py │ ├── prepare_module.py │ ├── progress.py │ ├── rank_zero_log.py │ ├── stateful.py │ ├── swa.py │ ├── test_utils.py │ ├── timer.py │ ├── tqdm.py │ └── version.py └── tox.ini /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.coveragerc -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/ISSUE_TEMPLATE/bug-report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/ISSUE_TEMPLATE/documentation.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/ISSUE_TEMPLATE/feature-request.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/help-support.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/ISSUE_TEMPLATE/help-support.yml -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/workflows/build_and_publish_docs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/workflows/build_and_publish_docs.yaml -------------------------------------------------------------------------------- /.github/workflows/build_docs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/workflows/build_docs.yaml -------------------------------------------------------------------------------- /.github/workflows/nightly_build_cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/workflows/nightly_build_cpu.yaml -------------------------------------------------------------------------------- /.github/workflows/pre_commit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/workflows/pre_commit.yaml -------------------------------------------------------------------------------- /.github/workflows/release_build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/workflows/release_build.yaml -------------------------------------------------------------------------------- /.github/workflows/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.github/workflows/test.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/README.md -------------------------------------------------------------------------------- /conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/conftest.py -------------------------------------------------------------------------------- /dev-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/dev-requirements.txt -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | src/pytorch-sphinx-theme/ 2 | -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/license_header.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/license_header.txt -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/_static/css/torchtnt.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/_static/css/torchtnt.css -------------------------------------------------------------------------------- /docs/source/_static/js/jquery.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/_static/js/jquery.js -------------------------------------------------------------------------------- /docs/source/_static/js/torchtnt.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/_static/js/torchtnt.js -------------------------------------------------------------------------------- /docs/source/assets/TNTDiagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/assets/TNTDiagram.png -------------------------------------------------------------------------------- /docs/source/checkpointing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/checkpointing.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/distributed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/distributed.rst -------------------------------------------------------------------------------- /docs/source/docutils.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/docutils.conf -------------------------------------------------------------------------------- /docs/source/examples.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/examples.rst -------------------------------------------------------------------------------- /docs/source/ext/fbcode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/ext/fbcode.py -------------------------------------------------------------------------------- /docs/source/framework/auto_unit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/auto_unit.rst -------------------------------------------------------------------------------- /docs/source/framework/callbacks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/callbacks.rst -------------------------------------------------------------------------------- /docs/source/framework/eval.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/eval.rst -------------------------------------------------------------------------------- /docs/source/framework/fit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/fit.rst -------------------------------------------------------------------------------- /docs/source/framework/predict.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/predict.rst -------------------------------------------------------------------------------- /docs/source/framework/state.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/state.rst -------------------------------------------------------------------------------- /docs/source/framework/train.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/train.rst -------------------------------------------------------------------------------- /docs/source/framework/unit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/framework/unit.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/overview.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/overview.rst -------------------------------------------------------------------------------- /docs/source/templates/class_template.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/templates/class_template.rst -------------------------------------------------------------------------------- /docs/source/templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/templates/layout.html -------------------------------------------------------------------------------- /docs/source/utils/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/docs/source/utils/utils.rst -------------------------------------------------------------------------------- /examples/auto_unit_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/auto_unit_example.py -------------------------------------------------------------------------------- /examples/mingpt/char_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/mingpt/char_dataset.py -------------------------------------------------------------------------------- /examples/mingpt/data/input.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/mingpt/data/input.txt -------------------------------------------------------------------------------- /examples/mingpt/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/mingpt/main.py -------------------------------------------------------------------------------- /examples/mingpt/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/mingpt/model.py -------------------------------------------------------------------------------- /examples/mnist/README.md: -------------------------------------------------------------------------------- 1 | # Basic MNIST Example 2 | 3 | ```bash 4 | pip install -r requirements.txt 5 | python main.py 6 | ``` 7 | -------------------------------------------------------------------------------- /examples/mnist/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/mnist/main.py -------------------------------------------------------------------------------- /examples/mnist/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/mnist/requirements.txt -------------------------------------------------------------------------------- /examples/torchrec/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/torchrec/README.md -------------------------------------------------------------------------------- /examples/torchrec/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/torchrec/main.py -------------------------------------------------------------------------------- /examples/torchrec/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/torchrec/requirements.txt -------------------------------------------------------------------------------- /examples/torchrec/tests/torchrec_example_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/torchrec/tests/torchrec_example_test.py -------------------------------------------------------------------------------- /examples/train_unit_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/examples/train_unit_example.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- 1 | [tool.usort] 2 | 3 | first_party_detection = false 4 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/setup.py -------------------------------------------------------------------------------- /tests/framework/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/__init__.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_base_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_base_checkpointer.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_checkpoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_checkpoint_utils.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_csv_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_csv_writer.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_dcp_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_dcp_saver.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_dcp_saver_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_dcp_saver_gpu.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_early_stopping.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_empty_cuda_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_empty_cuda_cache.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_empty_dataloader_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_empty_dataloader_detector.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_garbage_collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_garbage_collector.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_iteration_time_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_iteration_time_logger.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_lambda.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_learning_rate_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_learning_rate_monitor.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_memory_snapshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_memory_snapshot.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_module_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_module_summary.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_periodic_distributed_sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_periodic_distributed_sync.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_progress_reporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_progress_reporter.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_pytorch_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_pytorch_profiler.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_slow_rank_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_slow_rank_detector.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_system_resources_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_system_resources_monitor.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_tensorboard_parameter_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_tensorboard_parameter_monitor.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_tensorfloat32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_tensorfloat32.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_throughput_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_throughput_logger.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_time_limit_interrupter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_time_limit_interrupter.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_time_wait_for_batch_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_time_wait_for_batch_logger.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_torchsnapshot_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_torchsnapshot_saver.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_torchsnapshot_saver_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_torchsnapshot_saver_gpu.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_tqdm_progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_tqdm_progress_bar.py -------------------------------------------------------------------------------- /tests/framework/callbacks/test_train_progress_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/callbacks/test_train_progress_monitor.py -------------------------------------------------------------------------------- /tests/framework/test_app_state_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_app_state_mixin.py -------------------------------------------------------------------------------- /tests/framework/test_auto_unit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_auto_unit.py -------------------------------------------------------------------------------- /tests/framework/test_auto_unit_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_auto_unit_gpu.py -------------------------------------------------------------------------------- /tests/framework/test_callback_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_callback_handler.py -------------------------------------------------------------------------------- /tests/framework/test_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_evaluate.py -------------------------------------------------------------------------------- /tests/framework/test_fit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_fit.py -------------------------------------------------------------------------------- /tests/framework/test_loop_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_loop_utils.py -------------------------------------------------------------------------------- /tests/framework/test_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_predict.py -------------------------------------------------------------------------------- /tests/framework/test_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_state.py -------------------------------------------------------------------------------- /tests/framework/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_train.py -------------------------------------------------------------------------------- /tests/framework/test_unit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_unit.py -------------------------------------------------------------------------------- /tests/framework/test_unit_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_unit_utils.py -------------------------------------------------------------------------------- /tests/framework/test_unit_utils_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_unit_utils_gpu.py -------------------------------------------------------------------------------- /tests/framework/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/framework/test_utils.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/__init__.py -------------------------------------------------------------------------------- /tests/utils/data/test_data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/data/test_data_prefetcher.py -------------------------------------------------------------------------------- /tests/utils/data/test_data_prefetcher_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/data/test_data_prefetcher_gpu.py -------------------------------------------------------------------------------- /tests/utils/data/test_iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/data/test_iterators.py -------------------------------------------------------------------------------- /tests/utils/data/test_multi_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/data/test_multi_dataloader.py -------------------------------------------------------------------------------- /tests/utils/data/test_profile_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/data/test_profile_dataloader.py -------------------------------------------------------------------------------- /tests/utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/__init__.py -------------------------------------------------------------------------------- /tests/utils/loggers/test_anomaly_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/test_anomaly_logger.py -------------------------------------------------------------------------------- /tests/utils/loggers/test_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/test_csv.py -------------------------------------------------------------------------------- /tests/utils/loggers/test_in_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/test_in_memory.py -------------------------------------------------------------------------------- /tests/utils/loggers/test_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/test_json.py -------------------------------------------------------------------------------- /tests/utils/loggers/test_stdout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/test_stdout.py -------------------------------------------------------------------------------- /tests/utils/loggers/test_tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/test_tensorboard.py -------------------------------------------------------------------------------- /tests/utils/loggers/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/loggers/test_utils.py -------------------------------------------------------------------------------- /tests/utils/test_anomaly_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_anomaly_evaluation.py -------------------------------------------------------------------------------- /tests/utils/test_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_checkpoint.py -------------------------------------------------------------------------------- /tests/utils/test_checkpoint_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_checkpoint_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_device.py -------------------------------------------------------------------------------- /tests/utils/test_device_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_device_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_device_mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_device_mesh.py -------------------------------------------------------------------------------- /tests/utils/test_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_distributed.py -------------------------------------------------------------------------------- /tests/utils/test_distributed_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_distributed_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_early_stop_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_early_stop_checker.py -------------------------------------------------------------------------------- /tests/utils/test_early_stop_checker_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_early_stop_checker_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_env.py -------------------------------------------------------------------------------- /tests/utils/test_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_flops.py -------------------------------------------------------------------------------- /tests/utils/test_fsspec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_fsspec.py -------------------------------------------------------------------------------- /tests/utils/test_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_memory.py -------------------------------------------------------------------------------- /tests/utils/test_memory_snapshot_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_memory_snapshot_profiler.py -------------------------------------------------------------------------------- /tests/utils/test_memory_snapshot_profiler_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_memory_snapshot_profiler_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_misc.py -------------------------------------------------------------------------------- /tests/utils/test_module_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_module_summary.py -------------------------------------------------------------------------------- /tests/utils/test_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_nan.py -------------------------------------------------------------------------------- /tests/utils/test_oom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_oom.py -------------------------------------------------------------------------------- /tests/utils/test_oom_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_oom_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_optimizer.py -------------------------------------------------------------------------------- /tests/utils/test_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_precision.py -------------------------------------------------------------------------------- /tests/utils/test_prepare_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_prepare_module.py -------------------------------------------------------------------------------- /tests/utils/test_prepare_module_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_prepare_module_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_progress.py -------------------------------------------------------------------------------- /tests/utils/test_rank_zero_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_rank_zero_log.py -------------------------------------------------------------------------------- /tests/utils/test_swa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_swa.py -------------------------------------------------------------------------------- /tests/utils/test_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_timer.py -------------------------------------------------------------------------------- /tests/utils/test_timer_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_timer_gpu.py -------------------------------------------------------------------------------- /tests/utils/test_tqdm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_tqdm.py -------------------------------------------------------------------------------- /tests/utils/test_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tests/utils/test_version.py -------------------------------------------------------------------------------- /torchtnt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/__init__.py -------------------------------------------------------------------------------- /torchtnt/framework/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/__init__.py -------------------------------------------------------------------------------- /torchtnt/framework/_callback_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/_callback_handler.py -------------------------------------------------------------------------------- /torchtnt/framework/_loop_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/_loop_utils.py -------------------------------------------------------------------------------- /torchtnt/framework/_test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/_test_utils.py -------------------------------------------------------------------------------- /torchtnt/framework/_unit_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/_unit_utils.py -------------------------------------------------------------------------------- /torchtnt/framework/auto_unit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/auto_unit.py -------------------------------------------------------------------------------- /torchtnt/framework/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callback.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/__init__.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/_checkpoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/_checkpoint_utils.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/base_checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/base_checkpointer.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/base_csv_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/base_csv_writer.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/checkpointer_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/checkpointer_types.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/dcp_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/dcp_saver.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/early_stopping.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/empty_cuda_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/empty_cuda_cache.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/empty_dataloader_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/empty_dataloader_detector.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/garbage_collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/garbage_collector.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/iteration_time_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/iteration_time_logger.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/lambda_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/lambda_callback.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/learning_rate_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/learning_rate_monitor.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/memory_snapshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/memory_snapshot.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/module_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/module_summary.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/periodic_distributed_sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/periodic_distributed_sync.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/progress_reporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/progress_reporter.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/pytorch_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/pytorch_profiler.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/slow_rank_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/slow_rank_detector.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/system_resources_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/system_resources_monitor.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/tensorboard_parameter_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/tensorboard_parameter_monitor.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/tensorfloat32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/tensorfloat32.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/throughput_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/throughput_logger.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/time_limit_interrupter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/time_limit_interrupter.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/time_wait_for_batch_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/time_wait_for_batch_logger.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/torch_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/torch_compile.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/torchsnapshot_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/torchsnapshot_saver.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/tqdm_progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/tqdm_progress_bar.py -------------------------------------------------------------------------------- /torchtnt/framework/callbacks/train_progress_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/callbacks/train_progress_monitor.py -------------------------------------------------------------------------------- /torchtnt/framework/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/evaluate.py -------------------------------------------------------------------------------- /torchtnt/framework/fit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/fit.py -------------------------------------------------------------------------------- /torchtnt/framework/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/predict.py -------------------------------------------------------------------------------- /torchtnt/framework/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/state.py -------------------------------------------------------------------------------- /torchtnt/framework/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/train.py -------------------------------------------------------------------------------- /torchtnt/framework/unit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/unit.py -------------------------------------------------------------------------------- /torchtnt/framework/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/framework/utils.py -------------------------------------------------------------------------------- /torchtnt/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchtnt/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/__init__.py -------------------------------------------------------------------------------- /torchtnt/utils/anomaly_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/anomaly_evaluation.py -------------------------------------------------------------------------------- /torchtnt/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/checkpoint.py -------------------------------------------------------------------------------- /torchtnt/utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/data/__init__.py -------------------------------------------------------------------------------- /torchtnt/utils/data/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/data/data_prefetcher.py -------------------------------------------------------------------------------- /torchtnt/utils/data/iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/data/iterators.py -------------------------------------------------------------------------------- /torchtnt/utils/data/multi_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/data/multi_dataloader.py -------------------------------------------------------------------------------- /torchtnt/utils/data/profile_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/data/profile_dataloader.py -------------------------------------------------------------------------------- /torchtnt/utils/data/synthetic_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/data/synthetic_data.py -------------------------------------------------------------------------------- /torchtnt/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/device.py -------------------------------------------------------------------------------- /torchtnt/utils/device_mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/device_mesh.py -------------------------------------------------------------------------------- /torchtnt/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/distributed.py -------------------------------------------------------------------------------- /torchtnt/utils/early_stop_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/early_stop_checker.py -------------------------------------------------------------------------------- /torchtnt/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/env.py -------------------------------------------------------------------------------- /torchtnt/utils/event.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/event.py -------------------------------------------------------------------------------- /torchtnt/utils/event_handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/event_handlers.py -------------------------------------------------------------------------------- /torchtnt/utils/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/flops.py -------------------------------------------------------------------------------- /torchtnt/utils/fsdp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/fsdp_utils.py -------------------------------------------------------------------------------- /torchtnt/utils/fsspec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/fsspec.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/__init__.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/anomaly_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/anomaly_logger.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/csv.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/file.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/in_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/in_memory.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/json.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/logger.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/stdout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/stdout.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/tensorboard.py -------------------------------------------------------------------------------- /torchtnt/utils/loggers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/loggers/utils.py -------------------------------------------------------------------------------- /torchtnt/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/lr_scheduler.py -------------------------------------------------------------------------------- /torchtnt/utils/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/memory.py -------------------------------------------------------------------------------- /torchtnt/utils/memory_snapshot_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/memory_snapshot_profiler.py -------------------------------------------------------------------------------- /torchtnt/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/misc.py -------------------------------------------------------------------------------- /torchtnt/utils/module_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/module_summary.py -------------------------------------------------------------------------------- /torchtnt/utils/nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/nan.py -------------------------------------------------------------------------------- /torchtnt/utils/oom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/oom.py -------------------------------------------------------------------------------- /torchtnt/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/optimizer.py -------------------------------------------------------------------------------- /torchtnt/utils/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/precision.py -------------------------------------------------------------------------------- /torchtnt/utils/prepare_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/prepare_module.py -------------------------------------------------------------------------------- /torchtnt/utils/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/progress.py -------------------------------------------------------------------------------- /torchtnt/utils/rank_zero_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/rank_zero_log.py -------------------------------------------------------------------------------- /torchtnt/utils/stateful.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/stateful.py -------------------------------------------------------------------------------- /torchtnt/utils/swa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/swa.py -------------------------------------------------------------------------------- /torchtnt/utils/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/test_utils.py -------------------------------------------------------------------------------- /torchtnt/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/timer.py -------------------------------------------------------------------------------- /torchtnt/utils/tqdm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/tqdm.py -------------------------------------------------------------------------------- /torchtnt/utils/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/torchtnt/utils/version.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/tnt/HEAD/tox.ini --------------------------------------------------------------------------------