├── .git-blame-ignore-revs ├── .github ├── FUNDING.yml ├── ISSUE_TEMPLATE │ ├── bug-report.md │ ├── documentation.md │ ├── feature-request.md │ ├── questions-help-support.md │ └── user-feedback.md ├── PULL_REQUEST_TEMPLATE.md ├── dependabot.yml ├── failed_schedule_issue_template.md ├── pr-labeler-config.yml └── workflows │ ├── binaries-nightly-release.yml │ ├── build_docs.sh │ ├── code-style-checks.yml │ ├── discord_issues.yml │ ├── discord_pull_requests.yaml │ ├── docker-build.yml │ ├── docs.yml │ ├── gpu-hvd-tests.yml │ ├── gpu-tests.yml │ ├── hvd-tests.yml │ ├── install_docs_deps.sh │ ├── mps-tests.yml │ ├── pytorch-version-tests.yml │ ├── stable-release-anaconda.yml │ ├── stable-release-pypi.yml │ ├── tpu-tests.yml │ ├── triage.yml │ ├── typing-checks.yml │ └── unit-tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CITATION ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── assets ├── ignite_vs_bare_pytorch.png ├── logo │ ├── ignite_logo.png │ ├── ignite_logo.svg │ ├── ignite_logo_dark.png │ ├── ignite_logo_dark.svg │ ├── ignite_logo_guidelines.md │ ├── ignite_logo_light.png │ ├── ignite_logo_light.svg │ ├── ignite_logo_mixed.png │ ├── ignite_logo_mixed.svg │ ├── ignite_logo_mixed_light.png │ ├── ignite_logo_mixed_light.svg │ ├── ignite_logomark.png │ ├── ignite_logomark.svg │ ├── ignite_logomark_dark.png │ ├── ignite_logomark_dark.svg │ ├── ignite_logomark_light.png │ └── ignite_logomark_light.svg └── tldr │ ├── pytorch-ignite-teaser.gif │ ├── teaser.ipynb │ └── teaser.py ├── codecov.yml ├── conda.recipe ├── build_and_upload.sh └── meta.yaml ├── docker ├── README.md ├── build.sh ├── docker.cfg ├── hvd │ ├── Dockerfile.hvd-apex │ ├── Dockerfile.hvd-apex-nlp │ ├── Dockerfile.hvd-apex-vision │ ├── Dockerfile.hvd-base │ ├── Dockerfile.hvd-nlp │ └── Dockerfile.hvd-vision ├── main │ ├── Dockerfile.apex │ ├── Dockerfile.apex-nlp │ ├── Dockerfile.apex-vision │ ├── Dockerfile.base │ ├── Dockerfile.nlp │ └── Dockerfile.vision ├── msdp │ ├── Dockerfile.msdp-apex │ ├── Dockerfile.msdp-apex-nlp │ └── Dockerfile.msdp-apex-vision ├── push_all.sh └── test_image.py ├── docs ├── Makefile ├── make.bat ├── requirements.txt └── source │ ├── _static │ └── img │ │ ├── concepts │ │ └── timeline_and_events.png │ │ └── schedulers │ │ ├── concat_example.png │ │ ├── concat_linear_exp_step_lr.png │ │ ├── cosine_annealing_example.png │ │ ├── linear_cyclical_example.png │ │ ├── lr_scheduler.png │ │ ├── piecewise_linear.png │ │ └── reduce_lr_on_plateau_example.png │ ├── _templates │ ├── _static │ │ ├── css │ │ │ └── ignite_theme.css │ │ └── img │ │ │ ├── ignite_logo.svg │ │ │ ├── ignite_logomark.svg │ │ │ └── netlify-light.svg │ ├── autosummary │ │ ├── class.rst │ │ ├── function.rst │ │ └── module.rst │ ├── classwithcall.rst │ ├── fonts.html │ ├── footer.html │ ├── layout.html │ └── theme_variables.jinja │ ├── conf.py │ ├── contrib │ ├── engines.rst │ ├── handlers.rst │ └── metrics.rst │ ├── defaults.rst │ ├── distributed.rst │ ├── engine.rst │ ├── exceptions.rst │ ├── handlers.rst │ ├── index.rst │ ├── metrics.rst │ └── utils.rst ├── examples ├── cifar10 │ ├── .gitignore │ ├── README.md │ ├── main.py │ ├── requirements.txt │ └── utils.py ├── cifar100_amp_benchmark │ ├── benchmark_fp32.py │ ├── benchmark_nvidia_apex.py │ ├── benchmark_torch_cuda_amp.py │ └── utils.py ├── cifar10_qat │ ├── .gitignore │ ├── README.md │ ├── main.py │ ├── pact.py │ └── utils.py ├── fast_neural_style │ ├── README.md │ ├── handlers.py │ ├── images │ │ ├── content_images │ │ │ └── amber.jpg │ │ ├── output_images │ │ │ └── mosaic_amber.jpg │ │ └── style_images │ │ │ └── mosaic.jpg │ ├── neural_style.py │ ├── transformer_net.py │ ├── utils.py │ └── vgg.py ├── gan │ ├── README.md │ └── dcgan.py ├── mnist │ ├── README.md │ ├── assets │ │ ├── logs-det_run_1_2.png │ │ ├── logs-det_run_3_4.png │ │ ├── logs_run_1_2.png │ │ ├── logs_run_3_4.png │ │ ├── run_vs_resume_run_logs_1_2.png │ │ └── run_vs_resume_run_logs_3_4.png │ ├── mnist.py │ ├── mnist_save_resume_engine.py │ ├── mnist_with_clearml_logger.py │ ├── mnist_with_neptune_logger.py │ ├── mnist_with_tensorboard.py │ ├── mnist_with_tensorboard_logger.py │ ├── mnist_with_tensorboard_on_tpu.py │ ├── mnist_with_tqdm_logger.py │ ├── mnist_with_visdom.py │ ├── mnist_with_visdom_logger.py │ └── mnist_with_wandb_logger.py ├── notebooks │ ├── Cifar100_bench_amp.ipynb │ ├── Cifar10_Ax_hyperparam_tuning.ipynb │ ├── CycleGAN_with_nvidia_apex.ipynb │ ├── CycleGAN_with_torch_cuda_amp.ipynb │ ├── EfficientNet_Cifar100_finetuning.ipynb │ ├── FashionMNIST.ipynb │ ├── FastaiLRFinder_MNIST.ipynb │ ├── HandlersTimeProfiler_MNIST.ipynb │ ├── MNIST_on_TPU.ipynb │ ├── TextCNN.ipynb │ ├── VAE.ipynb │ └── assets │ │ ├── ax_hparams.png │ │ ├── efficientnets.png │ │ ├── fashion-mnist.png │ │ ├── fastresnet_v2.svg │ │ └── tb_cyclegan_images.png ├── references │ ├── README.md │ ├── classification │ │ └── imagenet │ │ │ ├── .gitignore │ │ │ ├── README.md │ │ │ ├── configs │ │ │ ├── baseline_resnet50.py │ │ │ └── check_baseline_resnet50.py │ │ │ ├── dataflow.py │ │ │ ├── main.py │ │ │ ├── requirements.txt │ │ │ ├── utils.py │ │ │ └── vis.py │ └── segmentation │ │ └── pascal_voc2012 │ │ ├── README.md │ │ ├── configs │ │ ├── baseline_dplv3_resnet101.py │ │ ├── baseline_dplv3_resnet101_sbd.py │ │ └── eval_baseline_dplv3_resnet101_sbd.py │ │ ├── dataflow.py │ │ ├── main.py │ │ ├── requirements.txt │ │ ├── utils.py │ │ └── vis.py ├── reinforcement_learning │ ├── README.md │ ├── actor_critic.py │ └── reinforce.py ├── siamese_network │ ├── README.md │ ├── requirements.txt │ └── siamese_network.py ├── super_resolution │ ├── README.md │ ├── images │ │ ├── bicubic_image_cifar.png │ │ ├── input_cifar.png │ │ └── out_cifar.png │ ├── main.py │ ├── model.py │ └── super_resolve.py └── transformers │ ├── README.md │ ├── dataset.py │ ├── main.py │ ├── model.py │ ├── requirements.txt │ └── utils.py ├── ignite ├── __init__.py ├── _utils.py ├── base │ ├── __init__.py │ └── mixins.py ├── contrib │ ├── __init__.py │ ├── engines │ │ ├── __init__.py │ │ ├── common.py │ │ └── tbptt.py │ ├── handlers │ │ ├── __init__.py │ │ ├── base_logger.py │ │ ├── clearml_logger.py │ │ ├── lr_finder.py │ │ ├── mlflow_logger.py │ │ ├── neptune_logger.py │ │ ├── param_scheduler.py │ │ ├── polyaxon_logger.py │ │ ├── tensorboard_logger.py │ │ ├── time_profilers.py │ │ ├── tqdm_logger.py │ │ ├── visdom_logger.py │ │ └── wandb_logger.py │ └── metrics │ │ ├── __init__.py │ │ ├── average_precision.py │ │ ├── cohen_kappa.py │ │ ├── gpu_info.py │ │ ├── precision_recall_curve.py │ │ ├── regression │ │ ├── __init__.py │ │ ├── _base.py │ │ ├── canberra_metric.py │ │ ├── fractional_absolute_error.py │ │ ├── fractional_bias.py │ │ ├── geometric_mean_absolute_error.py │ │ ├── geometric_mean_relative_absolute_error.py │ │ ├── manhattan_distance.py │ │ ├── maximum_absolute_error.py │ │ ├── mean_absolute_relative_error.py │ │ ├── mean_error.py │ │ ├── mean_normalized_bias.py │ │ ├── median_absolute_error.py │ │ ├── median_absolute_percentage_error.py │ │ ├── median_relative_absolute_error.py │ │ ├── r2_score.py │ │ └── wave_hedges_distance.py │ │ └── roc_auc.py ├── distributed │ ├── __init__.py │ ├── auto.py │ ├── comp_models │ │ ├── __init__.py │ │ ├── base.py │ │ ├── horovod.py │ │ ├── native.py │ │ └── xla.py │ ├── launcher.py │ └── utils.py ├── engine │ ├── __init__.py │ ├── deterministic.py │ ├── engine.py │ ├── events.py │ └── utils.py ├── exceptions.py ├── handlers │ ├── __init__.py │ ├── base_logger.py │ ├── checkpoint.py │ ├── clearml_logger.py │ ├── early_stopping.py │ ├── ema_handler.py │ ├── fbresearch_logger.py │ ├── lr_finder.py │ ├── mlflow_logger.py │ ├── neptune_logger.py │ ├── param_scheduler.py │ ├── polyaxon_logger.py │ ├── state_param_scheduler.py │ ├── stores.py │ ├── tensorboard_logger.py │ ├── terminate_on_nan.py │ ├── time_limit.py │ ├── time_profilers.py │ ├── timing.py │ ├── tqdm_logger.py │ ├── utils.py │ ├── visdom_logger.py │ └── wandb_logger.py ├── metrics │ ├── __init__.py │ ├── accumulation.py │ ├── accuracy.py │ ├── average_precision.py │ ├── classification_report.py │ ├── clustering │ │ ├── __init__.py │ │ ├── _base.py │ │ ├── calinski_harabasz_score.py │ │ ├── davies_bouldin_score.py │ │ └── silhouette_score.py │ ├── cohen_kappa.py │ ├── confusion_matrix.py │ ├── cosine_similarity.py │ ├── entropy.py │ ├── epoch_metric.py │ ├── fbeta.py │ ├── frequency.py │ ├── gan │ │ ├── __init__.py │ │ ├── fid.py │ │ ├── inception_score.py │ │ └── utils.py │ ├── gpu_info.py │ ├── hsic.py │ ├── js_divergence.py │ ├── kl_divergence.py │ ├── loss.py │ ├── maximum_mean_discrepancy.py │ ├── mean_absolute_error.py │ ├── mean_average_precision.py │ ├── mean_pairwise_distance.py │ ├── mean_squared_error.py │ ├── metric.py │ ├── metric_group.py │ ├── metrics_lambda.py │ ├── multilabel_confusion_matrix.py │ ├── mutual_information.py │ ├── nlp │ │ ├── __init__.py │ │ ├── bleu.py │ │ ├── rouge.py │ │ └── utils.py │ ├── precision.py │ ├── precision_recall_curve.py │ ├── psnr.py │ ├── recall.py │ ├── regression │ │ ├── __init__.py │ │ ├── _base.py │ │ ├── canberra_metric.py │ │ ├── fractional_absolute_error.py │ │ ├── fractional_bias.py │ │ ├── geometric_mean_absolute_error.py │ │ ├── geometric_mean_relative_absolute_error.py │ │ ├── kendall_correlation.py │ │ ├── manhattan_distance.py │ │ ├── maximum_absolute_error.py │ │ ├── mean_absolute_relative_error.py │ │ ├── mean_error.py │ │ ├── mean_normalized_bias.py │ │ ├── median_absolute_error.py │ │ ├── median_absolute_percentage_error.py │ │ ├── median_relative_absolute_error.py │ │ ├── pearson_correlation.py │ │ ├── r2_score.py │ │ ├── spearman_correlation.py │ │ └── wave_hedges_distance.py │ ├── roc_auc.py │ ├── root_mean_squared_error.py │ ├── running_average.py │ ├── ssim.py │ ├── top_k_categorical_accuracy.py │ └── vision │ │ ├── __init__.py │ │ └── object_detection_average_precision_recall.py ├── py.typed └── utils.py ├── mypy.ini ├── netlify.toml ├── pyproject.toml ├── pyrefly.toml ├── requirements-dev.txt └── tests ├── __init__.py ├── common_test_functionality.sh ├── ignite ├── __init__.py ├── base │ ├── __init__.py │ └── test_mixins.py ├── conftest.py ├── contrib │ ├── __init__.py │ ├── conftest.py │ ├── engines │ │ ├── __init__.py │ │ ├── test_common.py │ │ └── test_tbptt.py │ ├── handlers │ │ └── test_warnings_of_deprecation_of_handlers.py │ └── metrics │ │ └── test_warnings_of_deprecation_of_metrics.py ├── distributed │ ├── __init__.py │ ├── check_idist_parallel.py │ ├── comp_models │ │ ├── __init__.py │ │ ├── test_base.py │ │ ├── test_horovod.py │ │ ├── test_native.py │ │ └── test_xla.py │ ├── test_auto.py │ ├── test_launcher.py │ └── utils │ │ ├── __init__.py │ │ ├── test_horovod.py │ │ ├── test_native.py │ │ ├── test_serial.py │ │ └── test_xla.py ├── engine │ ├── __init__.py │ ├── test_create_supervised.py │ ├── test_custom_events.py │ ├── test_deterministic.py │ ├── test_engine.py │ ├── test_engine_state_dict.py │ ├── test_event_handlers.py │ └── test_memory_leaks.py ├── handlers │ ├── __init__.py │ ├── conftest.py │ ├── test_base_logger.py │ ├── test_checkpoint.py │ ├── test_clearml_logger.py │ ├── test_early_stopping.py │ ├── test_ema_handler.py │ ├── test_fbresearch_logger.py │ ├── test_handlers.py │ ├── test_lr_finder.py │ ├── test_mlflow_logger.py │ ├── test_neptune_logger.py │ ├── test_param_scheduler.py │ ├── test_polyaxon_logger.py │ ├── test_state_param_scheduler.py │ ├── test_stores.py │ ├── test_tensorboard_logger.py │ ├── test_terminate_on_nan.py │ ├── test_time_limit.py │ ├── test_time_profilers.py │ ├── test_timing.py │ ├── test_tqdm_logger.py │ ├── test_visdom_logger.py │ └── test_wandb_logger.py ├── metrics │ ├── __init__.py │ ├── clustering │ │ ├── __init__.py │ │ ├── test_calinski_harabasz_score.py │ │ ├── test_davies_bouldin_score.py │ │ └── test_silhouette_score.py │ ├── conftest.py │ ├── gan │ │ ├── __init__.py │ │ ├── test_fid.py │ │ ├── test_inception_score.py │ │ └── test_utils.py │ ├── nlp │ │ ├── __init__.py │ │ ├── test_bleu.py │ │ ├── test_rouge.py │ │ └── test_utils.py │ ├── regression │ │ ├── __init__.py │ │ ├── test__base.py │ │ ├── test_canberra_metric.py │ │ ├── test_fractional_absolute_error.py │ │ ├── test_fractional_bias.py │ │ ├── test_geometric_mean_absolute_error.py │ │ ├── test_geometric_mean_relative_absolute_error.py │ │ ├── test_kendall_correlation.py │ │ ├── test_manhattan_distance.py │ │ ├── test_maximum_absolute_error.py │ │ ├── test_mean_absolute_relative_error.py │ │ ├── test_mean_error.py │ │ ├── test_mean_normalized_bias.py │ │ ├── test_median_absolute_error.py │ │ ├── test_median_absolute_percentage_error.py │ │ ├── test_median_relative_absolute_error.py │ │ ├── test_pearson_correlation.py │ │ ├── test_r2_score.py │ │ ├── test_spearman_correlation.py │ │ └── test_wave_hedges_distance.py │ ├── test_accumulation.py │ ├── test_accuracy.py │ ├── test_average_precision.py │ ├── test_classification_report.py │ ├── test_cohen_kappa.py │ ├── test_confusion_matrix.py │ ├── test_cosine_similarity.py │ ├── test_dill.py │ ├── test_entropy.py │ ├── test_epoch_metric.py │ ├── test_fbeta.py │ ├── test_frequency.py │ ├── test_gpu_info.py │ ├── test_hsic.py │ ├── test_js_divergence.py │ ├── test_kl_divergence.py │ ├── test_loss.py │ ├── test_maximum_mean_discrepancy.py │ ├── test_mean_absolute_error.py │ ├── test_mean_average_precision.py │ ├── test_mean_pairwise_distance.py │ ├── test_mean_squared_error.py │ ├── test_metric.py │ ├── test_metric_group.py │ ├── test_metrics_lambda.py │ ├── test_multilabel_confusion_matrix.py │ ├── test_mutual_information.py │ ├── test_precision.py │ ├── test_precision_recall_curve.py │ ├── test_psnr.py │ ├── test_recall.py │ ├── test_roc_auc.py │ ├── test_roc_curve.py │ ├── test_root_mean_squared_error.py │ ├── test_running_average.py │ ├── test_ssim.py │ ├── test_top_k_categorical_accuracy.py │ └── vision │ │ ├── __init__.py │ │ └── test_object_detection_map.py └── test_utils.py ├── run_cpu_tests.sh ├── run_gpu_tests.sh ├── run_multinode_tests_in_docker.sh └── run_tpu_tests.sh /.git-blame-ignore-revs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.git-blame-ignore-revs -------------------------------------------------------------------------------- /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/ISSUE_TEMPLATE/bug-report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/ISSUE_TEMPLATE/documentation.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/ISSUE_TEMPLATE/feature-request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/questions-help-support.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/ISSUE_TEMPLATE/questions-help-support.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/user-feedback.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/ISSUE_TEMPLATE/user-feedback.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/failed_schedule_issue_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/failed_schedule_issue_template.md -------------------------------------------------------------------------------- /.github/pr-labeler-config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/pr-labeler-config.yml -------------------------------------------------------------------------------- /.github/workflows/binaries-nightly-release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/binaries-nightly-release.yml -------------------------------------------------------------------------------- /.github/workflows/build_docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/build_docs.sh -------------------------------------------------------------------------------- /.github/workflows/code-style-checks.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/code-style-checks.yml -------------------------------------------------------------------------------- /.github/workflows/discord_issues.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/discord_issues.yml -------------------------------------------------------------------------------- /.github/workflows/discord_pull_requests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/discord_pull_requests.yaml -------------------------------------------------------------------------------- /.github/workflows/docker-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/docker-build.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/gpu-hvd-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/gpu-hvd-tests.yml -------------------------------------------------------------------------------- /.github/workflows/gpu-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/gpu-tests.yml -------------------------------------------------------------------------------- /.github/workflows/hvd-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/hvd-tests.yml -------------------------------------------------------------------------------- /.github/workflows/install_docs_deps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/install_docs_deps.sh -------------------------------------------------------------------------------- /.github/workflows/mps-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/mps-tests.yml -------------------------------------------------------------------------------- /.github/workflows/pytorch-version-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/pytorch-version-tests.yml -------------------------------------------------------------------------------- /.github/workflows/stable-release-anaconda.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/stable-release-anaconda.yml -------------------------------------------------------------------------------- /.github/workflows/stable-release-pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/stable-release-pypi.yml -------------------------------------------------------------------------------- /.github/workflows/tpu-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/tpu-tests.yml -------------------------------------------------------------------------------- /.github/workflows/triage.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/triage.yml -------------------------------------------------------------------------------- /.github/workflows/typing-checks.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/typing-checks.yml -------------------------------------------------------------------------------- /.github/workflows/unit-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.github/workflows/unit-tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CITATION: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/CITATION -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/README.md -------------------------------------------------------------------------------- /assets/ignite_vs_bare_pytorch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/ignite_vs_bare_pytorch.png -------------------------------------------------------------------------------- /assets/logo/ignite_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo.png -------------------------------------------------------------------------------- /assets/logo/ignite_logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo.svg -------------------------------------------------------------------------------- /assets/logo/ignite_logo_dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_dark.png -------------------------------------------------------------------------------- /assets/logo/ignite_logo_dark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_dark.svg -------------------------------------------------------------------------------- /assets/logo/ignite_logo_guidelines.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_guidelines.md -------------------------------------------------------------------------------- /assets/logo/ignite_logo_light.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_light.png -------------------------------------------------------------------------------- /assets/logo/ignite_logo_light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_light.svg -------------------------------------------------------------------------------- /assets/logo/ignite_logo_mixed.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_mixed.png -------------------------------------------------------------------------------- /assets/logo/ignite_logo_mixed.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_mixed.svg -------------------------------------------------------------------------------- /assets/logo/ignite_logo_mixed_light.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_mixed_light.png -------------------------------------------------------------------------------- /assets/logo/ignite_logo_mixed_light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logo_mixed_light.svg -------------------------------------------------------------------------------- /assets/logo/ignite_logomark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logomark.png -------------------------------------------------------------------------------- /assets/logo/ignite_logomark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logomark.svg -------------------------------------------------------------------------------- /assets/logo/ignite_logomark_dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logomark_dark.png -------------------------------------------------------------------------------- /assets/logo/ignite_logomark_dark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logomark_dark.svg -------------------------------------------------------------------------------- /assets/logo/ignite_logomark_light.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logomark_light.png -------------------------------------------------------------------------------- /assets/logo/ignite_logomark_light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/logo/ignite_logomark_light.svg -------------------------------------------------------------------------------- /assets/tldr/pytorch-ignite-teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/tldr/pytorch-ignite-teaser.gif -------------------------------------------------------------------------------- /assets/tldr/teaser.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/tldr/teaser.ipynb -------------------------------------------------------------------------------- /assets/tldr/teaser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/assets/tldr/teaser.py -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/codecov.yml -------------------------------------------------------------------------------- /conda.recipe/build_and_upload.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/conda.recipe/build_and_upload.sh -------------------------------------------------------------------------------- /conda.recipe/meta.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/conda.recipe/meta.yaml -------------------------------------------------------------------------------- /docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/README.md -------------------------------------------------------------------------------- /docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/build.sh -------------------------------------------------------------------------------- /docker/docker.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/docker.cfg -------------------------------------------------------------------------------- /docker/hvd/Dockerfile.hvd-apex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/hvd/Dockerfile.hvd-apex -------------------------------------------------------------------------------- /docker/hvd/Dockerfile.hvd-apex-nlp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/hvd/Dockerfile.hvd-apex-nlp -------------------------------------------------------------------------------- /docker/hvd/Dockerfile.hvd-apex-vision: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/hvd/Dockerfile.hvd-apex-vision -------------------------------------------------------------------------------- /docker/hvd/Dockerfile.hvd-base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/hvd/Dockerfile.hvd-base -------------------------------------------------------------------------------- /docker/hvd/Dockerfile.hvd-nlp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/hvd/Dockerfile.hvd-nlp -------------------------------------------------------------------------------- /docker/hvd/Dockerfile.hvd-vision: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/hvd/Dockerfile.hvd-vision -------------------------------------------------------------------------------- /docker/main/Dockerfile.apex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/main/Dockerfile.apex -------------------------------------------------------------------------------- /docker/main/Dockerfile.apex-nlp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/main/Dockerfile.apex-nlp -------------------------------------------------------------------------------- /docker/main/Dockerfile.apex-vision: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/main/Dockerfile.apex-vision -------------------------------------------------------------------------------- /docker/main/Dockerfile.base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/main/Dockerfile.base -------------------------------------------------------------------------------- /docker/main/Dockerfile.nlp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/main/Dockerfile.nlp -------------------------------------------------------------------------------- /docker/main/Dockerfile.vision: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/main/Dockerfile.vision -------------------------------------------------------------------------------- /docker/msdp/Dockerfile.msdp-apex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/msdp/Dockerfile.msdp-apex -------------------------------------------------------------------------------- /docker/msdp/Dockerfile.msdp-apex-nlp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/msdp/Dockerfile.msdp-apex-nlp -------------------------------------------------------------------------------- /docker/msdp/Dockerfile.msdp-apex-vision: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/msdp/Dockerfile.msdp-apex-vision -------------------------------------------------------------------------------- /docker/push_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/push_all.sh -------------------------------------------------------------------------------- /docker/test_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docker/test_image.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/_static/img/concepts/timeline_and_events.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/concepts/timeline_and_events.png -------------------------------------------------------------------------------- /docs/source/_static/img/schedulers/concat_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/schedulers/concat_example.png -------------------------------------------------------------------------------- /docs/source/_static/img/schedulers/concat_linear_exp_step_lr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/schedulers/concat_linear_exp_step_lr.png -------------------------------------------------------------------------------- /docs/source/_static/img/schedulers/cosine_annealing_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/schedulers/cosine_annealing_example.png -------------------------------------------------------------------------------- /docs/source/_static/img/schedulers/linear_cyclical_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/schedulers/linear_cyclical_example.png -------------------------------------------------------------------------------- /docs/source/_static/img/schedulers/lr_scheduler.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/schedulers/lr_scheduler.png -------------------------------------------------------------------------------- /docs/source/_static/img/schedulers/piecewise_linear.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/schedulers/piecewise_linear.png -------------------------------------------------------------------------------- /docs/source/_static/img/schedulers/reduce_lr_on_plateau_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_static/img/schedulers/reduce_lr_on_plateau_example.png -------------------------------------------------------------------------------- /docs/source/_templates/_static/css/ignite_theme.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/_static/css/ignite_theme.css -------------------------------------------------------------------------------- /docs/source/_templates/_static/img/ignite_logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/_static/img/ignite_logo.svg -------------------------------------------------------------------------------- /docs/source/_templates/_static/img/ignite_logomark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/_static/img/ignite_logomark.svg -------------------------------------------------------------------------------- /docs/source/_templates/_static/img/netlify-light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/_static/img/netlify-light.svg -------------------------------------------------------------------------------- /docs/source/_templates/autosummary/class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/autosummary/class.rst -------------------------------------------------------------------------------- /docs/source/_templates/autosummary/function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/autosummary/function.rst -------------------------------------------------------------------------------- /docs/source/_templates/autosummary/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/autosummary/module.rst -------------------------------------------------------------------------------- /docs/source/_templates/classwithcall.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/classwithcall.rst -------------------------------------------------------------------------------- /docs/source/_templates/fonts.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/fonts.html -------------------------------------------------------------------------------- /docs/source/_templates/footer.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/footer.html -------------------------------------------------------------------------------- /docs/source/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/layout.html -------------------------------------------------------------------------------- /docs/source/_templates/theme_variables.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/_templates/theme_variables.jinja -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/contrib/engines.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/contrib/engines.rst -------------------------------------------------------------------------------- /docs/source/contrib/handlers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/contrib/handlers.rst -------------------------------------------------------------------------------- /docs/source/contrib/metrics.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/contrib/metrics.rst -------------------------------------------------------------------------------- /docs/source/defaults.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/defaults.rst -------------------------------------------------------------------------------- /docs/source/distributed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/distributed.rst -------------------------------------------------------------------------------- /docs/source/engine.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/engine.rst -------------------------------------------------------------------------------- /docs/source/exceptions.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/exceptions.rst -------------------------------------------------------------------------------- /docs/source/handlers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/handlers.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/metrics.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/metrics.rst -------------------------------------------------------------------------------- /docs/source/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/docs/source/utils.rst -------------------------------------------------------------------------------- /examples/cifar10/.gitignore: -------------------------------------------------------------------------------- 1 | output 2 | cifar10 3 | raw_pytorch 4 | -------------------------------------------------------------------------------- /examples/cifar10/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10/README.md -------------------------------------------------------------------------------- /examples/cifar10/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10/main.py -------------------------------------------------------------------------------- /examples/cifar10/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10/requirements.txt -------------------------------------------------------------------------------- /examples/cifar10/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10/utils.py -------------------------------------------------------------------------------- /examples/cifar100_amp_benchmark/benchmark_fp32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar100_amp_benchmark/benchmark_fp32.py -------------------------------------------------------------------------------- /examples/cifar100_amp_benchmark/benchmark_nvidia_apex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar100_amp_benchmark/benchmark_nvidia_apex.py -------------------------------------------------------------------------------- /examples/cifar100_amp_benchmark/benchmark_torch_cuda_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar100_amp_benchmark/benchmark_torch_cuda_amp.py -------------------------------------------------------------------------------- /examples/cifar100_amp_benchmark/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar100_amp_benchmark/utils.py -------------------------------------------------------------------------------- /examples/cifar10_qat/.gitignore: -------------------------------------------------------------------------------- 1 | output 2 | cifar10 3 | raw_pytorch 4 | -------------------------------------------------------------------------------- /examples/cifar10_qat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10_qat/README.md -------------------------------------------------------------------------------- /examples/cifar10_qat/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10_qat/main.py -------------------------------------------------------------------------------- /examples/cifar10_qat/pact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10_qat/pact.py -------------------------------------------------------------------------------- /examples/cifar10_qat/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/cifar10_qat/utils.py -------------------------------------------------------------------------------- /examples/fast_neural_style/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/README.md -------------------------------------------------------------------------------- /examples/fast_neural_style/handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/handlers.py -------------------------------------------------------------------------------- /examples/fast_neural_style/images/content_images/amber.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/images/content_images/amber.jpg -------------------------------------------------------------------------------- /examples/fast_neural_style/images/output_images/mosaic_amber.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/images/output_images/mosaic_amber.jpg -------------------------------------------------------------------------------- /examples/fast_neural_style/images/style_images/mosaic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/images/style_images/mosaic.jpg -------------------------------------------------------------------------------- /examples/fast_neural_style/neural_style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/neural_style.py -------------------------------------------------------------------------------- /examples/fast_neural_style/transformer_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/transformer_net.py -------------------------------------------------------------------------------- /examples/fast_neural_style/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/utils.py -------------------------------------------------------------------------------- /examples/fast_neural_style/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/fast_neural_style/vgg.py -------------------------------------------------------------------------------- /examples/gan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/gan/README.md -------------------------------------------------------------------------------- /examples/gan/dcgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/gan/dcgan.py -------------------------------------------------------------------------------- /examples/mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/README.md -------------------------------------------------------------------------------- /examples/mnist/assets/logs-det_run_1_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/assets/logs-det_run_1_2.png -------------------------------------------------------------------------------- /examples/mnist/assets/logs-det_run_3_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/assets/logs-det_run_3_4.png -------------------------------------------------------------------------------- /examples/mnist/assets/logs_run_1_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/assets/logs_run_1_2.png -------------------------------------------------------------------------------- /examples/mnist/assets/logs_run_3_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/assets/logs_run_3_4.png -------------------------------------------------------------------------------- /examples/mnist/assets/run_vs_resume_run_logs_1_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/assets/run_vs_resume_run_logs_1_2.png -------------------------------------------------------------------------------- /examples/mnist/assets/run_vs_resume_run_logs_3_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/assets/run_vs_resume_run_logs_3_4.png -------------------------------------------------------------------------------- /examples/mnist/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist.py -------------------------------------------------------------------------------- /examples/mnist/mnist_save_resume_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_save_resume_engine.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_clearml_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_clearml_logger.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_neptune_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_neptune_logger.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_tensorboard.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_tensorboard_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_tensorboard_logger.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_tensorboard_on_tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_tensorboard_on_tpu.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_tqdm_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_tqdm_logger.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_visdom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_visdom.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_visdom_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_visdom_logger.py -------------------------------------------------------------------------------- /examples/mnist/mnist_with_wandb_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/mnist/mnist_with_wandb_logger.py -------------------------------------------------------------------------------- /examples/notebooks/Cifar100_bench_amp.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/Cifar100_bench_amp.ipynb -------------------------------------------------------------------------------- /examples/notebooks/Cifar10_Ax_hyperparam_tuning.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/Cifar10_Ax_hyperparam_tuning.ipynb -------------------------------------------------------------------------------- /examples/notebooks/CycleGAN_with_nvidia_apex.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/CycleGAN_with_nvidia_apex.ipynb -------------------------------------------------------------------------------- /examples/notebooks/CycleGAN_with_torch_cuda_amp.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/CycleGAN_with_torch_cuda_amp.ipynb -------------------------------------------------------------------------------- /examples/notebooks/EfficientNet_Cifar100_finetuning.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/EfficientNet_Cifar100_finetuning.ipynb -------------------------------------------------------------------------------- /examples/notebooks/FashionMNIST.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/FashionMNIST.ipynb -------------------------------------------------------------------------------- /examples/notebooks/FastaiLRFinder_MNIST.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/FastaiLRFinder_MNIST.ipynb -------------------------------------------------------------------------------- /examples/notebooks/HandlersTimeProfiler_MNIST.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/HandlersTimeProfiler_MNIST.ipynb -------------------------------------------------------------------------------- /examples/notebooks/MNIST_on_TPU.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/MNIST_on_TPU.ipynb -------------------------------------------------------------------------------- /examples/notebooks/TextCNN.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/TextCNN.ipynb -------------------------------------------------------------------------------- /examples/notebooks/VAE.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/VAE.ipynb -------------------------------------------------------------------------------- /examples/notebooks/assets/ax_hparams.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/assets/ax_hparams.png -------------------------------------------------------------------------------- /examples/notebooks/assets/efficientnets.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/assets/efficientnets.png -------------------------------------------------------------------------------- /examples/notebooks/assets/fashion-mnist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/assets/fashion-mnist.png -------------------------------------------------------------------------------- /examples/notebooks/assets/fastresnet_v2.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/assets/fastresnet_v2.svg -------------------------------------------------------------------------------- /examples/notebooks/assets/tb_cyclegan_images.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/notebooks/assets/tb_cyclegan_images.png -------------------------------------------------------------------------------- /examples/references/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/README.md -------------------------------------------------------------------------------- /examples/references/classification/imagenet/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/.gitignore -------------------------------------------------------------------------------- /examples/references/classification/imagenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/README.md -------------------------------------------------------------------------------- /examples/references/classification/imagenet/configs/baseline_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/configs/baseline_resnet50.py -------------------------------------------------------------------------------- /examples/references/classification/imagenet/configs/check_baseline_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/configs/check_baseline_resnet50.py -------------------------------------------------------------------------------- /examples/references/classification/imagenet/dataflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/dataflow.py -------------------------------------------------------------------------------- /examples/references/classification/imagenet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/main.py -------------------------------------------------------------------------------- /examples/references/classification/imagenet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/requirements.txt -------------------------------------------------------------------------------- /examples/references/classification/imagenet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/utils.py -------------------------------------------------------------------------------- /examples/references/classification/imagenet/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/classification/imagenet/vis.py -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/README.md -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/configs/baseline_dplv3_resnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/configs/baseline_dplv3_resnet101.py -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/configs/baseline_dplv3_resnet101_sbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/configs/baseline_dplv3_resnet101_sbd.py -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/configs/eval_baseline_dplv3_resnet101_sbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/configs/eval_baseline_dplv3_resnet101_sbd.py -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/dataflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/dataflow.py -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/main.py -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/requirements.txt -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/utils.py -------------------------------------------------------------------------------- /examples/references/segmentation/pascal_voc2012/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/references/segmentation/pascal_voc2012/vis.py -------------------------------------------------------------------------------- /examples/reinforcement_learning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/reinforcement_learning/README.md -------------------------------------------------------------------------------- /examples/reinforcement_learning/actor_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/reinforcement_learning/actor_critic.py -------------------------------------------------------------------------------- /examples/reinforcement_learning/reinforce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/reinforcement_learning/reinforce.py -------------------------------------------------------------------------------- /examples/siamese_network/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/siamese_network/README.md -------------------------------------------------------------------------------- /examples/siamese_network/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/siamese_network/requirements.txt -------------------------------------------------------------------------------- /examples/siamese_network/siamese_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/siamese_network/siamese_network.py -------------------------------------------------------------------------------- /examples/super_resolution/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/super_resolution/README.md -------------------------------------------------------------------------------- /examples/super_resolution/images/bicubic_image_cifar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/super_resolution/images/bicubic_image_cifar.png -------------------------------------------------------------------------------- /examples/super_resolution/images/input_cifar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/super_resolution/images/input_cifar.png -------------------------------------------------------------------------------- /examples/super_resolution/images/out_cifar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/super_resolution/images/out_cifar.png -------------------------------------------------------------------------------- /examples/super_resolution/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/super_resolution/main.py -------------------------------------------------------------------------------- /examples/super_resolution/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/super_resolution/model.py -------------------------------------------------------------------------------- /examples/super_resolution/super_resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/super_resolution/super_resolve.py -------------------------------------------------------------------------------- /examples/transformers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/transformers/README.md -------------------------------------------------------------------------------- /examples/transformers/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/transformers/dataset.py -------------------------------------------------------------------------------- /examples/transformers/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/transformers/main.py -------------------------------------------------------------------------------- /examples/transformers/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/transformers/model.py -------------------------------------------------------------------------------- /examples/transformers/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/transformers/requirements.txt -------------------------------------------------------------------------------- /examples/transformers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/examples/transformers/utils.py -------------------------------------------------------------------------------- /ignite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/__init__.py -------------------------------------------------------------------------------- /ignite/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/_utils.py -------------------------------------------------------------------------------- /ignite/base/__init__.py: -------------------------------------------------------------------------------- 1 | from ignite.base.mixins import Serializable 2 | -------------------------------------------------------------------------------- /ignite/base/mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/base/mixins.py -------------------------------------------------------------------------------- /ignite/contrib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ignite/contrib/engines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/engines/__init__.py -------------------------------------------------------------------------------- /ignite/contrib/engines/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/engines/common.py -------------------------------------------------------------------------------- /ignite/contrib/engines/tbptt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/engines/tbptt.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/__init__.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/base_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/base_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/clearml_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/clearml_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/lr_finder.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/mlflow_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/mlflow_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/neptune_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/neptune_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/param_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/param_scheduler.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/polyaxon_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/polyaxon_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/tensorboard_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/tensorboard_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/time_profilers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/time_profilers.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/tqdm_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/tqdm_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/visdom_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/visdom_logger.py -------------------------------------------------------------------------------- /ignite/contrib/handlers/wandb_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/handlers/wandb_logger.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/__init__.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/average_precision.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/cohen_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/cohen_kappa.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/gpu_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/gpu_info.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/precision_recall_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/precision_recall_curve.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/__init__.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/_base.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/canberra_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/canberra_metric.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/fractional_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/fractional_absolute_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/fractional_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/fractional_bias.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/geometric_mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/geometric_mean_absolute_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/geometric_mean_relative_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/geometric_mean_relative_absolute_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/manhattan_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/manhattan_distance.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/maximum_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/maximum_absolute_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/mean_absolute_relative_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/mean_absolute_relative_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/mean_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/mean_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/mean_normalized_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/mean_normalized_bias.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/median_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/median_absolute_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/median_absolute_percentage_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/median_absolute_percentage_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/median_relative_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/median_relative_absolute_error.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/r2_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/r2_score.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/regression/wave_hedges_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/regression/wave_hedges_distance.py -------------------------------------------------------------------------------- /ignite/contrib/metrics/roc_auc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/contrib/metrics/roc_auc.py -------------------------------------------------------------------------------- /ignite/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/__init__.py -------------------------------------------------------------------------------- /ignite/distributed/auto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/auto.py -------------------------------------------------------------------------------- /ignite/distributed/comp_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/comp_models/__init__.py -------------------------------------------------------------------------------- /ignite/distributed/comp_models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/comp_models/base.py -------------------------------------------------------------------------------- /ignite/distributed/comp_models/horovod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/comp_models/horovod.py -------------------------------------------------------------------------------- /ignite/distributed/comp_models/native.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/comp_models/native.py -------------------------------------------------------------------------------- /ignite/distributed/comp_models/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/comp_models/xla.py -------------------------------------------------------------------------------- /ignite/distributed/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/launcher.py -------------------------------------------------------------------------------- /ignite/distributed/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/distributed/utils.py -------------------------------------------------------------------------------- /ignite/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/engine/__init__.py -------------------------------------------------------------------------------- /ignite/engine/deterministic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/engine/deterministic.py -------------------------------------------------------------------------------- /ignite/engine/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/engine/engine.py -------------------------------------------------------------------------------- /ignite/engine/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/engine/events.py -------------------------------------------------------------------------------- /ignite/engine/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/engine/utils.py -------------------------------------------------------------------------------- /ignite/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/exceptions.py -------------------------------------------------------------------------------- /ignite/handlers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/__init__.py -------------------------------------------------------------------------------- /ignite/handlers/base_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/base_logger.py -------------------------------------------------------------------------------- /ignite/handlers/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/checkpoint.py -------------------------------------------------------------------------------- /ignite/handlers/clearml_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/clearml_logger.py -------------------------------------------------------------------------------- /ignite/handlers/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/early_stopping.py -------------------------------------------------------------------------------- /ignite/handlers/ema_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/ema_handler.py -------------------------------------------------------------------------------- /ignite/handlers/fbresearch_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/fbresearch_logger.py -------------------------------------------------------------------------------- /ignite/handlers/lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/lr_finder.py -------------------------------------------------------------------------------- /ignite/handlers/mlflow_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/mlflow_logger.py -------------------------------------------------------------------------------- /ignite/handlers/neptune_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/neptune_logger.py -------------------------------------------------------------------------------- /ignite/handlers/param_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/param_scheduler.py -------------------------------------------------------------------------------- /ignite/handlers/polyaxon_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/polyaxon_logger.py -------------------------------------------------------------------------------- /ignite/handlers/state_param_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/state_param_scheduler.py -------------------------------------------------------------------------------- /ignite/handlers/stores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/stores.py -------------------------------------------------------------------------------- /ignite/handlers/tensorboard_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/tensorboard_logger.py -------------------------------------------------------------------------------- /ignite/handlers/terminate_on_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/terminate_on_nan.py -------------------------------------------------------------------------------- /ignite/handlers/time_limit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/time_limit.py -------------------------------------------------------------------------------- /ignite/handlers/time_profilers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/time_profilers.py -------------------------------------------------------------------------------- /ignite/handlers/timing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/timing.py -------------------------------------------------------------------------------- /ignite/handlers/tqdm_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/tqdm_logger.py -------------------------------------------------------------------------------- /ignite/handlers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/utils.py -------------------------------------------------------------------------------- /ignite/handlers/visdom_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/visdom_logger.py -------------------------------------------------------------------------------- /ignite/handlers/wandb_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/handlers/wandb_logger.py -------------------------------------------------------------------------------- /ignite/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/__init__.py -------------------------------------------------------------------------------- /ignite/metrics/accumulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/accumulation.py -------------------------------------------------------------------------------- /ignite/metrics/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/accuracy.py -------------------------------------------------------------------------------- /ignite/metrics/average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/average_precision.py -------------------------------------------------------------------------------- /ignite/metrics/classification_report.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/classification_report.py -------------------------------------------------------------------------------- /ignite/metrics/clustering/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/clustering/__init__.py -------------------------------------------------------------------------------- /ignite/metrics/clustering/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/clustering/_base.py -------------------------------------------------------------------------------- /ignite/metrics/clustering/calinski_harabasz_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/clustering/calinski_harabasz_score.py -------------------------------------------------------------------------------- /ignite/metrics/clustering/davies_bouldin_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/clustering/davies_bouldin_score.py -------------------------------------------------------------------------------- /ignite/metrics/clustering/silhouette_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/clustering/silhouette_score.py -------------------------------------------------------------------------------- /ignite/metrics/cohen_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/cohen_kappa.py -------------------------------------------------------------------------------- /ignite/metrics/confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/confusion_matrix.py -------------------------------------------------------------------------------- /ignite/metrics/cosine_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/cosine_similarity.py -------------------------------------------------------------------------------- /ignite/metrics/entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/entropy.py -------------------------------------------------------------------------------- /ignite/metrics/epoch_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/epoch_metric.py -------------------------------------------------------------------------------- /ignite/metrics/fbeta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/fbeta.py -------------------------------------------------------------------------------- /ignite/metrics/frequency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/frequency.py -------------------------------------------------------------------------------- /ignite/metrics/gan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/gan/__init__.py -------------------------------------------------------------------------------- /ignite/metrics/gan/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/gan/fid.py -------------------------------------------------------------------------------- /ignite/metrics/gan/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/gan/inception_score.py -------------------------------------------------------------------------------- /ignite/metrics/gan/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/gan/utils.py -------------------------------------------------------------------------------- /ignite/metrics/gpu_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/gpu_info.py -------------------------------------------------------------------------------- /ignite/metrics/hsic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/hsic.py -------------------------------------------------------------------------------- /ignite/metrics/js_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/js_divergence.py -------------------------------------------------------------------------------- /ignite/metrics/kl_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/kl_divergence.py -------------------------------------------------------------------------------- /ignite/metrics/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/loss.py -------------------------------------------------------------------------------- /ignite/metrics/maximum_mean_discrepancy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/maximum_mean_discrepancy.py -------------------------------------------------------------------------------- /ignite/metrics/mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/mean_absolute_error.py -------------------------------------------------------------------------------- /ignite/metrics/mean_average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/mean_average_precision.py -------------------------------------------------------------------------------- /ignite/metrics/mean_pairwise_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/mean_pairwise_distance.py -------------------------------------------------------------------------------- /ignite/metrics/mean_squared_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/mean_squared_error.py -------------------------------------------------------------------------------- /ignite/metrics/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/metric.py -------------------------------------------------------------------------------- /ignite/metrics/metric_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/metric_group.py -------------------------------------------------------------------------------- /ignite/metrics/metrics_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/metrics_lambda.py -------------------------------------------------------------------------------- /ignite/metrics/multilabel_confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/multilabel_confusion_matrix.py -------------------------------------------------------------------------------- /ignite/metrics/mutual_information.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/mutual_information.py -------------------------------------------------------------------------------- /ignite/metrics/nlp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/nlp/__init__.py -------------------------------------------------------------------------------- /ignite/metrics/nlp/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/nlp/bleu.py -------------------------------------------------------------------------------- /ignite/metrics/nlp/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/nlp/rouge.py -------------------------------------------------------------------------------- /ignite/metrics/nlp/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/nlp/utils.py -------------------------------------------------------------------------------- /ignite/metrics/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/precision.py -------------------------------------------------------------------------------- /ignite/metrics/precision_recall_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/precision_recall_curve.py -------------------------------------------------------------------------------- /ignite/metrics/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/psnr.py -------------------------------------------------------------------------------- /ignite/metrics/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/recall.py -------------------------------------------------------------------------------- /ignite/metrics/regression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/__init__.py -------------------------------------------------------------------------------- /ignite/metrics/regression/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/_base.py -------------------------------------------------------------------------------- /ignite/metrics/regression/canberra_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/canberra_metric.py -------------------------------------------------------------------------------- /ignite/metrics/regression/fractional_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/fractional_absolute_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/fractional_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/fractional_bias.py -------------------------------------------------------------------------------- /ignite/metrics/regression/geometric_mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/geometric_mean_absolute_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/geometric_mean_relative_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/geometric_mean_relative_absolute_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/kendall_correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/kendall_correlation.py -------------------------------------------------------------------------------- /ignite/metrics/regression/manhattan_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/manhattan_distance.py -------------------------------------------------------------------------------- /ignite/metrics/regression/maximum_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/maximum_absolute_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/mean_absolute_relative_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/mean_absolute_relative_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/mean_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/mean_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/mean_normalized_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/mean_normalized_bias.py -------------------------------------------------------------------------------- /ignite/metrics/regression/median_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/median_absolute_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/median_absolute_percentage_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/median_absolute_percentage_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/median_relative_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/median_relative_absolute_error.py -------------------------------------------------------------------------------- /ignite/metrics/regression/pearson_correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/pearson_correlation.py -------------------------------------------------------------------------------- /ignite/metrics/regression/r2_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/r2_score.py -------------------------------------------------------------------------------- /ignite/metrics/regression/spearman_correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/spearman_correlation.py -------------------------------------------------------------------------------- /ignite/metrics/regression/wave_hedges_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/regression/wave_hedges_distance.py -------------------------------------------------------------------------------- /ignite/metrics/roc_auc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/roc_auc.py -------------------------------------------------------------------------------- /ignite/metrics/root_mean_squared_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/root_mean_squared_error.py -------------------------------------------------------------------------------- /ignite/metrics/running_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/running_average.py -------------------------------------------------------------------------------- /ignite/metrics/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/ssim.py -------------------------------------------------------------------------------- /ignite/metrics/top_k_categorical_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/top_k_categorical_accuracy.py -------------------------------------------------------------------------------- /ignite/metrics/vision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/vision/__init__.py -------------------------------------------------------------------------------- /ignite/metrics/vision/object_detection_average_precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/metrics/vision/object_detection_average_precision_recall.py -------------------------------------------------------------------------------- /ignite/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ignite/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/ignite/utils.py -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/mypy.ini -------------------------------------------------------------------------------- /netlify.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/netlify.toml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pyrefly.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/pyrefly.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/requirements-dev.txt -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | # Needed to collect coverage data 2 | -------------------------------------------------------------------------------- /tests/common_test_functionality.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/common_test_functionality.sh -------------------------------------------------------------------------------- /tests/ignite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/__init__.py -------------------------------------------------------------------------------- /tests/ignite/base/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/base/test_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/base/test_mixins.py -------------------------------------------------------------------------------- /tests/ignite/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/conftest.py -------------------------------------------------------------------------------- /tests/ignite/contrib/__init__.py: -------------------------------------------------------------------------------- 1 | # coding: utf-8 2 | -------------------------------------------------------------------------------- /tests/ignite/contrib/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/contrib/conftest.py -------------------------------------------------------------------------------- /tests/ignite/contrib/engines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/contrib/engines/test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/contrib/engines/test_common.py -------------------------------------------------------------------------------- /tests/ignite/contrib/engines/test_tbptt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/contrib/engines/test_tbptt.py -------------------------------------------------------------------------------- /tests/ignite/contrib/handlers/test_warnings_of_deprecation_of_handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/contrib/handlers/test_warnings_of_deprecation_of_handlers.py -------------------------------------------------------------------------------- /tests/ignite/contrib/metrics/test_warnings_of_deprecation_of_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/contrib/metrics/test_warnings_of_deprecation_of_metrics.py -------------------------------------------------------------------------------- /tests/ignite/distributed/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/distributed/check_idist_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/check_idist_parallel.py -------------------------------------------------------------------------------- /tests/ignite/distributed/comp_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/distributed/comp_models/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/comp_models/test_base.py -------------------------------------------------------------------------------- /tests/ignite/distributed/comp_models/test_horovod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/comp_models/test_horovod.py -------------------------------------------------------------------------------- /tests/ignite/distributed/comp_models/test_native.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/comp_models/test_native.py -------------------------------------------------------------------------------- /tests/ignite/distributed/comp_models/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/comp_models/test_xla.py -------------------------------------------------------------------------------- /tests/ignite/distributed/test_auto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/test_auto.py -------------------------------------------------------------------------------- /tests/ignite/distributed/test_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/test_launcher.py -------------------------------------------------------------------------------- /tests/ignite/distributed/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/utils/__init__.py -------------------------------------------------------------------------------- /tests/ignite/distributed/utils/test_horovod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/utils/test_horovod.py -------------------------------------------------------------------------------- /tests/ignite/distributed/utils/test_native.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/utils/test_native.py -------------------------------------------------------------------------------- /tests/ignite/distributed/utils/test_serial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/utils/test_serial.py -------------------------------------------------------------------------------- /tests/ignite/distributed/utils/test_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/distributed/utils/test_xla.py -------------------------------------------------------------------------------- /tests/ignite/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/__init__.py -------------------------------------------------------------------------------- /tests/ignite/engine/test_create_supervised.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/test_create_supervised.py -------------------------------------------------------------------------------- /tests/ignite/engine/test_custom_events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/test_custom_events.py -------------------------------------------------------------------------------- /tests/ignite/engine/test_deterministic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/test_deterministic.py -------------------------------------------------------------------------------- /tests/ignite/engine/test_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/test_engine.py -------------------------------------------------------------------------------- /tests/ignite/engine/test_engine_state_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/test_engine_state_dict.py -------------------------------------------------------------------------------- /tests/ignite/engine/test_event_handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/test_event_handlers.py -------------------------------------------------------------------------------- /tests/ignite/engine/test_memory_leaks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/engine/test_memory_leaks.py -------------------------------------------------------------------------------- /tests/ignite/handlers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/__init__.py -------------------------------------------------------------------------------- /tests/ignite/handlers/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/conftest.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_base_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_base_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_checkpoint.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_clearml_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_clearml_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_early_stopping.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_ema_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_ema_handler.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_fbresearch_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_fbresearch_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_handlers.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_lr_finder.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_mlflow_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_mlflow_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_neptune_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_neptune_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_param_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_param_scheduler.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_polyaxon_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_polyaxon_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_state_param_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_state_param_scheduler.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_stores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_stores.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_tensorboard_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_tensorboard_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_terminate_on_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_terminate_on_nan.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_time_limit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_time_limit.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_time_profilers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_time_profilers.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_timing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_timing.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_tqdm_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_tqdm_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_visdom_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_visdom_logger.py -------------------------------------------------------------------------------- /tests/ignite/handlers/test_wandb_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/handlers/test_wandb_logger.py -------------------------------------------------------------------------------- /tests/ignite/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | # Needed to collect coverage data 2 | -------------------------------------------------------------------------------- /tests/ignite/metrics/clustering/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/metrics/clustering/test_calinski_harabasz_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/clustering/test_calinski_harabasz_score.py -------------------------------------------------------------------------------- /tests/ignite/metrics/clustering/test_davies_bouldin_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/clustering/test_davies_bouldin_score.py -------------------------------------------------------------------------------- /tests/ignite/metrics/clustering/test_silhouette_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/clustering/test_silhouette_score.py -------------------------------------------------------------------------------- /tests/ignite/metrics/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/conftest.py -------------------------------------------------------------------------------- /tests/ignite/metrics/gan/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/metrics/gan/test_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/gan/test_fid.py -------------------------------------------------------------------------------- /tests/ignite/metrics/gan/test_inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/gan/test_inception_score.py -------------------------------------------------------------------------------- /tests/ignite/metrics/gan/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/gan/test_utils.py -------------------------------------------------------------------------------- /tests/ignite/metrics/nlp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/nlp/__init__.py -------------------------------------------------------------------------------- /tests/ignite/metrics/nlp/test_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/nlp/test_bleu.py -------------------------------------------------------------------------------- /tests/ignite/metrics/nlp/test_rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/nlp/test_rouge.py -------------------------------------------------------------------------------- /tests/ignite/metrics/nlp/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/nlp/test_utils.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test__base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test__base.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_canberra_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_canberra_metric.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_fractional_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_fractional_absolute_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_fractional_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_fractional_bias.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_geometric_mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_geometric_mean_absolute_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_geometric_mean_relative_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_geometric_mean_relative_absolute_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_kendall_correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_kendall_correlation.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_manhattan_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_manhattan_distance.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_maximum_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_maximum_absolute_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_mean_absolute_relative_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_mean_absolute_relative_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_mean_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_mean_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_mean_normalized_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_mean_normalized_bias.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_median_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_median_absolute_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_median_absolute_percentage_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_median_absolute_percentage_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_median_relative_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_median_relative_absolute_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_pearson_correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_pearson_correlation.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_r2_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_r2_score.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_spearman_correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_spearman_correlation.py -------------------------------------------------------------------------------- /tests/ignite/metrics/regression/test_wave_hedges_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/regression/test_wave_hedges_distance.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_accumulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_accumulation.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_accuracy.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_average_precision.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_classification_report.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_classification_report.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_cohen_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_cohen_kappa.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_confusion_matrix.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_cosine_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_cosine_similarity.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_dill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_dill.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_entropy.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_epoch_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_epoch_metric.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_fbeta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_fbeta.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_frequency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_frequency.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_gpu_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_gpu_info.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_hsic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_hsic.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_js_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_js_divergence.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_kl_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_kl_divergence.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_loss.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_maximum_mean_discrepancy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_maximum_mean_discrepancy.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_mean_absolute_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_mean_average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_mean_average_precision.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_mean_pairwise_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_mean_pairwise_distance.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_mean_squared_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_mean_squared_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_metric.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_metric_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_metric_group.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_metrics_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_metrics_lambda.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_multilabel_confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_multilabel_confusion_matrix.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_mutual_information.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_mutual_information.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_precision.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_precision_recall_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_precision_recall_curve.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_psnr.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_recall.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_roc_auc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_roc_auc.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_roc_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_roc_curve.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_root_mean_squared_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_root_mean_squared_error.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_running_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_running_average.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_ssim.py -------------------------------------------------------------------------------- /tests/ignite/metrics/test_top_k_categorical_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/test_top_k_categorical_accuracy.py -------------------------------------------------------------------------------- /tests/ignite/metrics/vision/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ignite/metrics/vision/test_object_detection_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/metrics/vision/test_object_detection_map.py -------------------------------------------------------------------------------- /tests/ignite/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/ignite/test_utils.py -------------------------------------------------------------------------------- /tests/run_cpu_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/run_cpu_tests.sh -------------------------------------------------------------------------------- /tests/run_gpu_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/run_gpu_tests.sh -------------------------------------------------------------------------------- /tests/run_multinode_tests_in_docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/run_multinode_tests_in_docker.sh -------------------------------------------------------------------------------- /tests/run_tpu_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/ignite/HEAD/tests/run_tpu_tests.sh --------------------------------------------------------------------------------