├── .github └── workflows │ ├── deploy-docs.yml │ ├── run-tests.yml │ └── update_docs_citations.yml ├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── docs ├── Makefile ├── _static │ ├── css │ │ ├── custom.css │ │ └── docstring_custom.css │ ├── images │ │ ├── examples │ │ │ ├── bert_horizonal.png │ │ │ ├── imagenet_corrupted.png │ │ │ ├── it3_mnist.png │ │ │ ├── mnist_noisy.png │ │ │ └── ttt_schema.png │ │ ├── teaser.svg │ │ ├── teaser_ssl_code.svg │ │ ├── teaser_ssl_schema.svg │ │ └── torch-ttt.svg │ └── js │ │ └── theme.js ├── _templates │ ├── class.rst │ ├── gallery_footer.html │ └── layout.html ├── api.rst ├── conf.py ├── index.rst ├── installation.rst ├── make.bat ├── papers.rst ├── quickstart.rst ├── references.bib ├── scripts │ ├── data │ │ ├── papers.yaml │ │ ├── papers_table.html │ │ └── papers_with_citations.yaml │ ├── generate_papers_table.py │ ├── push_examples.sh │ ├── ttt_citations_graph.py │ ├── ttt_papers.yaml │ └── update_papers.py └── source │ └── _static │ └── images │ └── torch-ttt.png ├── examples ├── 01_mnist_TTT.py ├── 02_imagenet_tent.py ├── 03_mae_bert.py ├── 04_mnist_it3.py └── GALLERY_HEADER.rst ├── pyproject.toml ├── requirements └── requirements.txt ├── tests ├── __init__.py ├── engines │ ├── __init__.py │ ├── test_actmad_engine.py │ ├── test_deyo_engine.py │ ├── test_eata_engine.py │ ├── test_it3_engine.py │ ├── test_masked_ttt_engine.py │ ├── test_memo_engine.py │ ├── test_ttt_engine.py │ └── test_ttt_pp_engine.py └── test_imports.py └── torch_ttt ├── __init__.py ├── engine ├── __init__.py ├── actmad_engine.py ├── base_engine.py ├── deyo_engine.py ├── eata_engine.py ├── it3_engine.py ├── masked_ttt_engine.py ├── memo_engine.py ├── tent_engine.py ├── ttt_engine.py └── ttt_pp_engine.py ├── engine_registry.py ├── loss ├── __init__.py ├── base_loss.py ├── contrastive_loss.py ├── entropy_loss.py ├── mean_loss.py ├── ttt_loss.py ├── weights_magnitude_loss.py └── zerot_loss.py ├── loss_registry.py └── utils ├── __init__.py ├── augmentations.py └── math.py /.github/workflows/deploy-docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/.github/workflows/deploy-docs.yml -------------------------------------------------------------------------------- /.github/workflows/run-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/.github/workflows/run-tests.yml -------------------------------------------------------------------------------- /.github/workflows/update_docs_citations.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/.github/workflows/update_docs_citations.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/README.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/css/custom.css -------------------------------------------------------------------------------- /docs/_static/css/docstring_custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/css/docstring_custom.css -------------------------------------------------------------------------------- /docs/_static/images/examples/bert_horizonal.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/examples/bert_horizonal.png -------------------------------------------------------------------------------- /docs/_static/images/examples/imagenet_corrupted.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/examples/imagenet_corrupted.png -------------------------------------------------------------------------------- /docs/_static/images/examples/it3_mnist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/examples/it3_mnist.png -------------------------------------------------------------------------------- /docs/_static/images/examples/mnist_noisy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/examples/mnist_noisy.png -------------------------------------------------------------------------------- /docs/_static/images/examples/ttt_schema.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/examples/ttt_schema.png -------------------------------------------------------------------------------- /docs/_static/images/teaser.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/teaser.svg -------------------------------------------------------------------------------- /docs/_static/images/teaser_ssl_code.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/teaser_ssl_code.svg -------------------------------------------------------------------------------- /docs/_static/images/teaser_ssl_schema.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/teaser_ssl_schema.svg -------------------------------------------------------------------------------- /docs/_static/images/torch-ttt.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/images/torch-ttt.svg -------------------------------------------------------------------------------- /docs/_static/js/theme.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_static/js/theme.js -------------------------------------------------------------------------------- /docs/_templates/class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_templates/class.rst -------------------------------------------------------------------------------- /docs/_templates/gallery_footer.html: -------------------------------------------------------------------------------- 1 | {# This template overrides the footer, leaving it empty #} -------------------------------------------------------------------------------- /docs/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/_templates/layout.html -------------------------------------------------------------------------------- /docs/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/api.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/installation.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/papers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/papers.rst -------------------------------------------------------------------------------- /docs/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/quickstart.rst -------------------------------------------------------------------------------- /docs/references.bib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/references.bib -------------------------------------------------------------------------------- /docs/scripts/data/papers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/data/papers.yaml -------------------------------------------------------------------------------- /docs/scripts/data/papers_table.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/data/papers_table.html -------------------------------------------------------------------------------- /docs/scripts/data/papers_with_citations.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/data/papers_with_citations.yaml -------------------------------------------------------------------------------- /docs/scripts/generate_papers_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/generate_papers_table.py -------------------------------------------------------------------------------- /docs/scripts/push_examples.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/push_examples.sh -------------------------------------------------------------------------------- /docs/scripts/ttt_citations_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/ttt_citations_graph.py -------------------------------------------------------------------------------- /docs/scripts/ttt_papers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/ttt_papers.yaml -------------------------------------------------------------------------------- /docs/scripts/update_papers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/scripts/update_papers.py -------------------------------------------------------------------------------- /docs/source/_static/images/torch-ttt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/docs/source/_static/images/torch-ttt.png -------------------------------------------------------------------------------- /examples/01_mnist_TTT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/examples/01_mnist_TTT.py -------------------------------------------------------------------------------- /examples/02_imagenet_tent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/examples/02_imagenet_tent.py -------------------------------------------------------------------------------- /examples/03_mae_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/examples/03_mae_bert.py -------------------------------------------------------------------------------- /examples/04_mnist_it3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/examples/04_mnist_it3.py -------------------------------------------------------------------------------- /examples/GALLERY_HEADER.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/examples/GALLERY_HEADER.rst -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements/requirements.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/engines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/engines/test_actmad_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_actmad_engine.py -------------------------------------------------------------------------------- /tests/engines/test_deyo_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_deyo_engine.py -------------------------------------------------------------------------------- /tests/engines/test_eata_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_eata_engine.py -------------------------------------------------------------------------------- /tests/engines/test_it3_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_it3_engine.py -------------------------------------------------------------------------------- /tests/engines/test_masked_ttt_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_masked_ttt_engine.py -------------------------------------------------------------------------------- /tests/engines/test_memo_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_memo_engine.py -------------------------------------------------------------------------------- /tests/engines/test_ttt_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_ttt_engine.py -------------------------------------------------------------------------------- /tests/engines/test_ttt_pp_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/engines/test_ttt_pp_engine.py -------------------------------------------------------------------------------- /tests/test_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/tests/test_imports.py -------------------------------------------------------------------------------- /torch_ttt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch_ttt/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/__init__.py -------------------------------------------------------------------------------- /torch_ttt/engine/actmad_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/actmad_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/base_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/base_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/deyo_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/deyo_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/eata_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/eata_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/it3_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/it3_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/masked_ttt_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/masked_ttt_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/memo_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/memo_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/tent_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/tent_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/ttt_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/ttt_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine/ttt_pp_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine/ttt_pp_engine.py -------------------------------------------------------------------------------- /torch_ttt/engine_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/engine_registry.py -------------------------------------------------------------------------------- /torch_ttt/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/__init__.py -------------------------------------------------------------------------------- /torch_ttt/loss/base_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/base_loss.py -------------------------------------------------------------------------------- /torch_ttt/loss/contrastive_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/contrastive_loss.py -------------------------------------------------------------------------------- /torch_ttt/loss/entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/entropy_loss.py -------------------------------------------------------------------------------- /torch_ttt/loss/mean_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/mean_loss.py -------------------------------------------------------------------------------- /torch_ttt/loss/ttt_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/ttt_loss.py -------------------------------------------------------------------------------- /torch_ttt/loss/weights_magnitude_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/weights_magnitude_loss.py -------------------------------------------------------------------------------- /torch_ttt/loss/zerot_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss/zerot_loss.py -------------------------------------------------------------------------------- /torch_ttt/loss_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/loss_registry.py -------------------------------------------------------------------------------- /torch_ttt/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch_ttt/utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/utils/augmentations.py -------------------------------------------------------------------------------- /torch_ttt/utils/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikitadurasov/torch-ttt/HEAD/torch_ttt/utils/math.py --------------------------------------------------------------------------------