.gitattributes
.gitignore
.pre-commit-config.yaml
CHANGELOG.md
CONTRIBUTING.md
LICENSE
Makefile
NOTICE
README.md
pyproject.toml
.github/copilot-instructions.md
.github/pull_request_template.md
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/docs_improvement.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/ISSUE_TEMPLATE/question.md
.github/ISSUE_TEMPLATE/setup_help.md
.github/workflows/check_code_format.yml
.github/workflows/test_build.yml
.github/workflows/test_documentation.yml
.github/workflows/test_unit.yml
.github/workflows/test_unit_minimal_dependencies.yml
dev_tools/dinov2_licenseheader.tmpl
dev_tools/dinov3_licenseheader.tmpl
dev_tools/eomt_licenseheader.tmpl
dev_tools/licenseheader.tmpl
dev_tools/pytorch_lightning_licenseheader.tmpl
docker/Dockerfile-amd64-cuda
docker/Makefile
docker/README.md
docker/variables.mk
docs/Makefile
docs/build.py
docs/format_code.py
docs/prebuild.py
docs/source/conf.py
docs/source/docker.md
docs/source/embed.md
docs/source/export.md
docs/source/faq.md
docs/source/index.md
docs/source/installation.md
docs/source/quick_start.md
docs/source/semantic_segmentation.md
docs/source/_static/custom.css
docs/source/_static/lightly_train_dark.svg
docs/source/_static/lightly_train_light.svg
docs/source/_static/images/tutorials/embedding/umap_lightly_train.jpg
docs/source/_static/images/tutorials/embedding/umap_lightly_train_colored.jpg
docs/source/_static/images/tutorials/embedding/umap_lightly_train_imagenet_colored.jpg
docs/source/_templates/page.html
docs/source/_templates/gtm/body.html
docs/source/_templates/gtm/head.html
docs/source/_templates/sidebar/version.html
docs/source/methods/dino.md
docs/source/methods/dinov2.md
docs/source/methods/distillation.md
docs/source/methods/index.md
docs/source/methods/simclr.md
docs/source/models/custom_models.md
docs/source/models/index.md
docs/source/models/rfdetr.md
docs/source/models/rtdetr.md
docs/source/models/supergradients.md
docs/source/models/timm.md
docs/source/models/torchvision.md
docs/source/models/ultralytics.md
docs/source/models/yolov12.md
docs/source/performance/hardware_recommendations.md
docs/source/performance/index.md
docs/source/performance/multi_gpu.md
docs/source/performance/multi_node.md
docs/source/python_api/index.md
docs/source/python_api/lightly_train.md
docs/source/train/index.md
docs/source/train/method_transform_args.md
docs/source/tutorials/index.md
docs/source/tutorials/colab/index.md
docs/source/tutorials/depth_estimation/diode_outdoor_samples.png
docs/source/tutorials/depth_estimation/index.md
docs/source/tutorials/depth_estimation/megadepth_samples.png
docs/source/tutorials/embedding/index.md
docs/source/tutorials/resnet/human-detection-dataset-fine-tuning.jpg
docs/source/tutorials/resnet/human-detection-dataset-pretraining.jpg
docs/source/tutorials/resnet/index.md
docs/source/tutorials/yolo/index.md
docs/source/tutorials/yolo/results_VOC.png
docs/source/tutorials/yolo/samples_VOC_train2012.png
examples/notebooks/quick_start.ipynb
examples/notebooks/rfdetr.ipynb
examples/notebooks/torchvision_embedding_model.ipynb
examples/notebooks/ultralytics_yolo.ipynb
examples/notebooks/yolov12.ipynb
licences/DINOv2_LICENSE
licences/DINOv3_LICENSE.md
licences/EOMT_LICENSE
licences/PYTORCH_LIGHTNING_LICENSE
src/lightly_train/__init__.py
src/lightly_train/_checkpoint.py
src/lightly_train/_cli.py
src/lightly_train/_distributed.py
src/lightly_train/_env.py
src/lightly_train/_float32_matmul_precision.py
src/lightly_train/_lightning_rank_zero.py
src/lightly_train/_logging.py
src/lightly_train/_plot.py
src/lightly_train/_scaling.py
src/lightly_train/_system.py
src/lightly_train/_task_checkpoint.py
src/lightly_train/_train_task_state.py
src/lightly_train/errors.py
src/lightly_train/model_wrappers.py
src/lightly_train/py.typed
src/lightly_train/types.py
src/lightly_train.egg-info/PKG-INFO
src/lightly_train.egg-info/SOURCES.txt
src/lightly_train.egg-info/dependency_links.txt
src/lightly_train.egg-info/entry_points.txt
src/lightly_train.egg-info/requires.txt
src/lightly_train.egg-info/top_level.txt
src/lightly_train/_callbacks/__init__.py
src/lightly_train/_callbacks/callback_args.py
src/lightly_train/_callbacks/callback_helpers.py
src/lightly_train/_callbacks/checkpoint.py
src/lightly_train/_callbacks/export.py
src/lightly_train/_callbacks/mlflow_logging.py
src/lightly_train/_callbacks/tqdm_progress_bar.py
src/lightly_train/_commands/__init__.py
src/lightly_train/_commands/_warnings.py
src/lightly_train/_commands/common_helpers.py
src/lightly_train/_commands/embed.py
src/lightly_train/_commands/export.py
src/lightly_train/_commands/export_task.py
src/lightly_train/_commands/extract_video_frames.py
src/lightly_train/_commands/train.py
src/lightly_train/_commands/train_helpers.py
src/lightly_train/_commands/train_task.py
src/lightly_train/_commands/train_task_helpers.py
src/lightly_train/_configs/__init__.py
src/lightly_train/_configs/config.py
src/lightly_train/_configs/omegaconf_utils.py
src/lightly_train/_configs/validate.py
src/lightly_train/_data/__init__.py
src/lightly_train/_data/cache.py
src/lightly_train/_data/download.py
src/lightly_train/_data/file_helpers.py
src/lightly_train/_data/image_dataset.py
src/lightly_train/_data/infinite_cycle_iterator.py
src/lightly_train/_data/mask_semantic_segmentation_dataset.py
src/lightly_train/_data/task_data_args.py
src/lightly_train/_data/yolo_object_detection_dataset.py
src/lightly_train/_data/_serialize/__init__.py
src/lightly_train/_data/_serialize/memory_mapped_sequence.py
src/lightly_train/_embedding/__init__.py
src/lightly_train/_embedding/embedding_format.py
src/lightly_train/_embedding/embedding_predictor.py
src/lightly_train/_embedding/embedding_transform.py
src/lightly_train/_embedding/writers/__init__.py
src/lightly_train/_embedding/writers/csv_writer.py
src/lightly_train/_embedding/writers/embedding_writer.py
src/lightly_train/_embedding/writers/torch_writer.py
src/lightly_train/_embedding/writers/writer_helpers.py
src/lightly_train/_loggers/__init__.py
src/lightly_train/_loggers/jsonl.py
src/lightly_train/_loggers/logger_args.py
src/lightly_train/_loggers/logger_helpers.py
src/lightly_train/_loggers/mlflow.py
src/lightly_train/_loggers/task_logger_args.py
src/lightly_train/_loggers/tensorboard.py
src/lightly_train/_loggers/wandb.py
src/lightly_train/_methods/__init__.py
src/lightly_train/_methods/method.py
src/lightly_train/_methods/method_args.py
src/lightly_train/_methods/method_helpers.py
src/lightly_train/_methods/densecl/__init__.py
src/lightly_train/_methods/densecl/densecl.py
src/lightly_train/_methods/densecl/densecl_loss.py
src/lightly_train/_methods/densecl/densecl_transform.py
src/lightly_train/_methods/detcon/detcon.py
src/lightly_train/_methods/detcon/detcon_transform.py
src/lightly_train/_methods/dino/__init__.py
src/lightly_train/_methods/dino/dino.py
src/lightly_train/_methods/dino/dino_transform.py
src/lightly_train/_methods/dinov2/__init__.py
src/lightly_train/_methods/dinov2/dinov2.py
src/lightly_train/_methods/dinov2/dinov2_head.py
src/lightly_train/_methods/dinov2/dinov2_loss.py
src/lightly_train/_methods/dinov2/dinov2_transform.py
src/lightly_train/_methods/dinov2/scheduler.py
src/lightly_train/_methods/dinov2/utils.py
src/lightly_train/_methods/distillation/__init__.py
src/lightly_train/_methods/distillation/distillation.py
src/lightly_train/_methods/distillation/distillation_loss.py
src/lightly_train/_methods/distillation/distillation_transform.py
src/lightly_train/_methods/distillationv2/__init__.py
src/lightly_train/_methods/distillationv2/distillationv2.py
src/lightly_train/_methods/distillationv2/distillationv2_loss.py
src/lightly_train/_methods/distillationv2/distillationv2_transform.py
src/lightly_train/_methods/simclr/__init__.py
src/lightly_train/_methods/simclr/simclr.py
src/lightly_train/_methods/simclr/simclr_transform.py
src/lightly_train/_models/__init__.py
src/lightly_train/_models/embedding_model.py
src/lightly_train/_models/log_usage_example.py
src/lightly_train/_models/model_wrapper.py
src/lightly_train/_models/package.py
src/lightly_train/_models/package_helpers.py
src/lightly_train/_models/custom/__init__.py
src/lightly_train/_models/custom/custom_package.py
src/lightly_train/_models/dinov2_vit/__init__.py
src/lightly_train/_models/dinov2_vit/dinov2_vit.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_package.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/dinov2_helper.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/__init__.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/ssl_default_config.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vitb14_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vitb14_reg4_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vitg14_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vitg14_reg4_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vitl14_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vitl14_reg4_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vits14_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/eval/vits14_reg4_pretrain.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/_vittest14.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vitb14.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vitb14_reg4.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vitg14.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vitg14_reg4.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vitl14.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vitl14_reg4.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vits14.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/configs/train/vits14_reg4.yaml
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/__init__.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/attention.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/block.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/drop_path.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/layer_scale.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/mlp.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/patch_embed.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/layers/swiglu_ffn.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/models/__init__.py
src/lightly_train/_models/dinov2_vit/dinov2_vit_src/models/vision_transformer.py
src/lightly_train/_models/dinov3/__init__.py
src/lightly_train/_models/dinov3/dinov3_package.py
src/lightly_train/_models/dinov3/dinov3_vit.py
src/lightly_train/_models/dinov3/dinov3_src/__init__.py
src/lightly_train/_models/dinov3/dinov3_src/hub/__init__.py
src/lightly_train/_models/dinov3/dinov3_src/hub/backbones.py
src/lightly_train/_models/dinov3/dinov3_src/hub/utils.py
src/lightly_train/_models/dinov3/dinov3_src/layers/__init__.py
src/lightly_train/_models/dinov3/dinov3_src/layers/attention.py
src/lightly_train/_models/dinov3/dinov3_src/layers/block.py
src/lightly_train/_models/dinov3/dinov3_src/layers/dino_head.py
src/lightly_train/_models/dinov3/dinov3_src/layers/ffn_layers.py
src/lightly_train/_models/dinov3/dinov3_src/layers/fp8_linear.py
src/lightly_train/_models/dinov3/dinov3_src/layers/layer_scale.py
src/lightly_train/_models/dinov3/dinov3_src/layers/patch_embed.py
src/lightly_train/_models/dinov3/dinov3_src/layers/rms_norm.py
src/lightly_train/_models/dinov3/dinov3_src/layers/rope_position_encoding.py
src/lightly_train/_models/dinov3/dinov3_src/layers/sparse_linear.py
src/lightly_train/_models/dinov3/dinov3_src/models/__init__.py
src/lightly_train/_models/dinov3/dinov3_src/models/convnext.py
src/lightly_train/_models/dinov3/dinov3_src/models/vision_transformer.py
src/lightly_train/_models/dinov3/dinov3_src/utils/__init__.py
src/lightly_train/_models/dinov3/dinov3_src/utils/cluster.py
src/lightly_train/_models/dinov3/dinov3_src/utils/custom_callable.py
src/lightly_train/_models/dinov3/dinov3_src/utils/dtype.py
src/lightly_train/_models/dinov3/dinov3_src/utils/utils.py
src/lightly_train/_models/rfdetr/__init__.py
src/lightly_train/_models/rfdetr/rfdetr.py
src/lightly_train/_models/rfdetr/rfdetr_package.py
src/lightly_train/_models/rtdetr/rtdetr.py
src/lightly_train/_models/super_gradients/__init__.py
src/lightly_train/_models/super_gradients/customizable_detector.py
src/lightly_train/_models/super_gradients/segmentation_module.py
src/lightly_train/_models/super_gradients/super_gradients.py
src/lightly_train/_models/super_gradients/super_gradients_package.py
src/lightly_train/_models/timm/__init__.py
src/lightly_train/_models/timm/timm.py
src/lightly_train/_models/timm/timm_package.py
src/lightly_train/_models/torchvision/__init__.py
src/lightly_train/_models/torchvision/convnext.py
src/lightly_train/_models/torchvision/resnet.py
src/lightly_train/_models/torchvision/shufflenet.py
src/lightly_train/_models/torchvision/torchvision.py
src/lightly_train/_models/torchvision/torchvision_package.py
src/lightly_train/_models/ultralytics/__init__.py
src/lightly_train/_models/ultralytics/ultralytics.py
src/lightly_train/_models/ultralytics/ultralytics_package.py
src/lightly_train/_optim/__init__.py
src/lightly_train/_optim/adamw_args.py
src/lightly_train/_optim/lars_args.py
src/lightly_train/_optim/optimizer_args.py
src/lightly_train/_optim/optimizer_helpers.py
src/lightly_train/_optim/optimizer_type.py
src/lightly_train/_optim/sgd_args.py
src/lightly_train/_optim/trainable_modules.py
src/lightly_train/_task_models/__init__.py
src/lightly_train/_task_models/task_model.py
src/lightly_train/_task_models/task_model_helpers.py
src/lightly_train/_task_models/train_model.py
src/lightly_train/_task_models/dinov2_eomt_semantic_segmentation/__init__.py
src/lightly_train/_task_models/dinov2_eomt_semantic_segmentation/mask_loss.py
src/lightly_train/_task_models/dinov2_eomt_semantic_segmentation/scale_block.py
src/lightly_train/_task_models/dinov2_eomt_semantic_segmentation/scheduler.py
src/lightly_train/_task_models/dinov2_eomt_semantic_segmentation/task_model.py
src/lightly_train/_task_models/dinov2_eomt_semantic_segmentation/train_model.py
src/lightly_train/_task_models/dinov2_eomt_semantic_segmentation/transforms.py
src/lightly_train/_task_models/dinov2_linear_semantic_segmentation/task_model.py
src/lightly_train/_task_models/dinov2_linear_semantic_segmentation/train_model.py
src/lightly_train/_task_models/dinov2_linear_semantic_segmentation/transforms.py
src/lightly_train/_task_models/dinov3_eomt_semantic_segmentation/__init__.py
src/lightly_train/_task_models/dinov3_eomt_semantic_segmentation/mask_loss.py
src/lightly_train/_task_models/dinov3_eomt_semantic_segmentation/scale_block.py
src/lightly_train/_task_models/dinov3_eomt_semantic_segmentation/scheduler.py
src/lightly_train/_task_models/dinov3_eomt_semantic_segmentation/task_model.py
src/lightly_train/_task_models/dinov3_eomt_semantic_segmentation/train_model.py
src/lightly_train/_task_models/dinov3_eomt_semantic_segmentation/transforms.py
src/lightly_train/_transforms/__init__.py
src/lightly_train/_transforms/channel_drop.py
src/lightly_train/_transforms/object_detection_transform.py
src/lightly_train/_transforms/semantic_segmentation_transform.py
src/lightly_train/_transforms/task_transform.py
src/lightly_train/_transforms/transform.py
src/lightly_train/_transforms/view_transform.py
tests/__init__.py
tests/helpers.py
tests/test__checkpoint.py
tests/test__cli.py
tests/test__env.py
tests/test__float32_matmul_precision.py
tests/test__logging.py
tests/test__plot.py
tests/test__scaling.py
tests/test__system.py
tests/_callbacks/__init__.py
tests/_callbacks/test_callback_helpers.py
tests/_commands/__init__.py
tests/_commands/test_common_helpers.py
tests/_commands/test_embed.py
tests/_commands/test_export.py
tests/_commands/test_export_task.py
tests/_commands/test_extract_video_frames.py
tests/_commands/test_train.py
tests/_commands/test_train_helpers.py
tests/_commands/test_train_task.py
tests/_configs/__init__.py
tests/_configs/test_config.py
tests/_configs/test_validate.py
tests/_data/__init__.py
tests/_data/test_cache.py
tests/_data/test_image_dataset.py
tests/_data/test_mask_semantic_segmentation_dataset.py
tests/_data/test_yolo_object_detection_dataset.py
tests/_data/_serialize/test_memory_mapped_sequence.py
tests/_embedding/__init__.py
tests/_embedding/writers/__init__.py
tests/_embedding/writers/test_csv_writer.py
tests/_embedding/writers/test_torch_writer.py
tests/_embedding/writers/test_writer_helpers.py
tests/_loggers/test_logger_helpers.py
tests/_methods/__init__.py
tests/_methods/test_method.py
tests/_methods/test_method_args.py
tests/_methods/test_method_helpers.py
tests/_methods/densecl/__init__.py
tests/_methods/densecl/test_densecl.py
tests/_methods/densecl/test_densecl_transform.py
tests/_methods/detcon/__init__.py
tests/_methods/detcon/test_detcon.py
tests/_methods/detcon/test_detcon_transform.py
tests/_methods/dino/__init__.py
tests/_methods/dino/test_dino.py
tests/_methods/dino/test_dino_transform.py
tests/_methods/dinov2/__init__.py
tests/_methods/dinov2/test_dinov2.py
tests/_methods/dinov2/test_dinov2_loss.py
tests/_methods/dinov2/test_dinov2_transform.py
tests/_methods/dinov2/test_utils.py
tests/_methods/distillation/__init__.py
tests/_methods/distillation/test_distillation.py
tests/_methods/distillation/test_distillation_loss.py
tests/_methods/distillation/test_distillation_transform.py
tests/_methods/distillationv2/__init__.py
tests/_methods/distillationv2/test_distillationv2.py
tests/_methods/distillationv2/test_distillationv2_loss.py
tests/_methods/distillationv2/test_distillationv2_transforms.py
tests/_methods/simclr/__init__.py
tests/_methods/simclr/test_simclr.py
tests/_methods/simclr/test_simclr_transform.py
tests/_models/__init__.py
tests/_models/test_embedding_model.py
tests/_models/test_model_wrapper.py
tests/_models/test_package_helpers.py
tests/_models/custom/__init__.py
tests/_models/custom/test_custom_package.py
tests/_models/dinov2_vit/__init__.py
tests/_models/dinov2_vit/test_dinov2_vit.py
tests/_models/dinov2_vit/test_dinov2_vit_package.py
tests/_models/rfdetr/__init__.py
tests/_models/rfdetr/test_rfdetr.py
tests/_models/rfdetr/test_rfdetr_package.py
tests/_models/super_gradients/__init__.py
tests/_models/super_gradients/test_segmentation_module.py
tests/_models/super_gradients/test_super_gradients_package.py
tests/_models/timm/__init__.py
tests/_models/timm/test_timm.py
tests/_models/timm/test_timm_package.py
tests/_models/torchvision/__init__.py
tests/_models/torchvision/test_convnext.py
tests/_models/torchvision/test_resnet.py
tests/_models/torchvision/test_shufflenet.py
tests/_models/torchvision/test_torchvision_package.py
tests/_models/ultralytics/test_ultralytics.py
tests/_models/ultralytics/test_ultralytics_package.py
tests/_optim/__init__.py
tests/_optim/test_optimizer_helpers.py
tests/_transforms/__init__.py
tests/_transforms/test_channel_drop.py
tests/_transforms/test_view_transform.py