.gitignore
CONTRIBUTE.md
LICENSE
Makefile
README.md
pyproject.toml
.github/workflows/release_pypi.yml
.github/workflows/test.yml
.github/workflows/test_code_format.yml
.github/workflows/test_docker.yml
.github/workflows/test_minimal_deps.yml
.github/workflows/weekly_dependency_tests.yml
docker/Dockerfile-amd64-cuda
docker/Makefile
docker/README.md
src/lightly_train/__init__.py
src/lightly_train/_checkpoint.py
src/lightly_train/_cli.py
src/lightly_train/_constants.py
src/lightly_train/_scaling.py
src/lightly_train/_transforms.py
src/lightly_train/errors.py
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/checkpoint.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/extract_video_frames.py
src/lightly_train/_commands/train.py
src/lightly_train/_commands/train_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/_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/tensorboard.py
src/lightly_train/_methods/__init__.py
src/lightly_train/_methods/densecl.py
src/lightly_train/_methods/densecldino.py
src/lightly_train/_methods/dino.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/simclr.py
src/lightly_train/_models/__init__.py
src/lightly_train/_models/embedding_model.py
src/lightly_train/_models/feature_extractor.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.py
src/lightly_train/_models/custom/custom_package.py
src/lightly_train/_models/embedding/__init__.py
src/lightly_train/_models/embedding/base.py
src/lightly_train/_models/super_gradients/__init__.py
src/lightly_train/_models/super_gradients/customizable_detector.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/torchvision.py
src/lightly_train/_models/torchvision/torchvision_package.py
src/lightly_train/_optim/__init__.py
src/lightly_train/_optim/adamw_args.py
src/lightly_train/_optim/optimizer.py
src/lightly_train/_optim/optimizer_args.py
src/lightly_train/_optim/optimizer_type.py
src/lightly_train/_optim/trainable_modules.py
tests/__init__.py
tests/helpers.py
tests/test__checkpoint.py
tests/test__cli.py
tests/test__scaling.py
tests/_commands/__init__.py
tests/_commands/test_common_helpers.py
tests/_commands/test_embed.py
tests/_commands/test_export.py
tests/_commands/test_extract_video_frames.py
tests/_commands/test_train.py
tests/_commands/test_train_helpers.py
tests/_configs/__init__.py
tests/_configs/test_validate.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/_methods/__init__.py
tests/_methods/test_method_helpers.py
tests/_models/__init__.py
tests/_models/test_package_helpers.py
tests/_models/custom/__init__.py
tests/_models/custom/test_custom.py
tests/_models/custom/test_custom_package.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/test_torchvision_package.py
tests/_optim/__init__.py
tests/_optim/test_optimizer.py