Metadata-Version: 2.4
Name: lightly_train
Version: 0.6.0
Summary: Train models with self-supervised learning in a single command
Author: Lightly Team
License: AGPL-3.0
Keywords: machine-learning,computer-vision,deep-learning,self-supervised-learning,contrastive-learning,pytorch,python,pretrained-models,embeddings
Requires-Python: >=3.8
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: albumentations>=1.3.1
Requires-Dist: eval-type-backport>=0.2
Requires-Dist: lightly>=1.5.16
Requires-Dist: omegaconf>=2.3
Requires-Dist: psutil>=5.0
Requires-Dist: pyarrow>=15.0.0
Requires-Dist: pytorch_lightning>=2.1
Requires-Dist: torch>=2.1.0
Requires-Dist: torchvision>=0.16.0
Requires-Dist: tqdm>=4.0.0
Requires-Dist: transformers>=4.40
Provides-Extra: dev
Requires-Dist: licenseheaders>=0.8.8; extra == "dev"
Requires-Dist: mdformat>=0.7.21; python_version >= "3.9" and extra == "dev"
Requires-Dist: mypy>=1.10.0; extra == "dev"
Requires-Dist: pre-commit>=3.5.0; extra == "dev"
Requires-Dist: pytest-examples>=0.0.13; extra == "dev"
Requires-Dist: pytest-mock>=3.14.0; extra == "dev"
Requires-Dist: pytest>=8.0.0; extra == "dev"
Requires-Dist: ruff>=0.4.0; extra == "dev"
Requires-Dist: types-pillow>=10.2.0; extra == "dev"
Requires-Dist: types-tqdm>=4.0.0; extra == "dev"
Requires-Dist: furo>=2024.04.27; extra == "dev"
Requires-Dist: myst_parser>=3.0; extra == "dev"
Requires-Dist: sphinx-autoapi>=3.0; extra == "dev"
Requires-Dist: sphinx-copybutton>=0.5.2; extra == "dev"
Requires-Dist: sphinx-design>=0.6.1; python_version >= "3.9" and extra == "dev"
Requires-Dist: sphinx-inline-tabs>=2023.4.21; extra == "dev"
Requires-Dist: sphinxcontrib-googleanalytics>=0.4; extra == "dev"
Requires-Dist: sphinx>=7.1; extra == "dev"
Provides-Extra: notebook
Requires-Dist: notebook>=7.2.3; extra == "notebook"
Provides-Extra: super-gradients
Requires-Dist: super-gradients>=3.7; extra == "super-gradients"
Provides-Extra: tensorboard
Requires-Dist: tensorboard>=2.10.0; extra == "tensorboard"
Provides-Extra: timm
Requires-Dist: timm>=1.0.3; extra == "timm"
Provides-Extra: ultralytics
Requires-Dist: ultralytics>=8.1.0; extra == "ultralytics"
Provides-Extra: wandb
Requires-Dist: wandb>=0.12.10; extra == "wandb"
Dynamic: license-file

# LightlyTrain

*Train models with self-supervised learning in a single command*

## Why LightlyTrain

Lightly**Train** uses self-supervised learning (SSL) to train models on large datasets
without the need for labels. It provides simple Python, Command Line, and Docker
interfaces to train models with popular SSL methods such as SimCLR or DINO. The trained
models are ideal starting points for fine-tuning on downstream tasks such as image
classification, object detection, and segmentation or for generating image embeddings.
Models trained with Lightly**Train** result in improved performance, faster convergence, and
better generalization compared to models trained without SSL. Image embeddings created
with Lightly**Train** capture more relevant information than their supervised counterparts
and seamlessly extend to new classes due to the unsupervised nature of SSL.

Lightly is the expert in SSL for computer vision and developed Lightly**Train** to simplify
model training for any task and dataset.

## Features

- Train models on any image data without labels
- Train models from popular libraries such as [torchvision](https://github.com/pytorch/vision), [TIMM](https://github.com/huggingface/pytorch-image-models), [Ultralytics](https://github.com/ultralytics/ultralytics), and [SuperGradients](https://github.com/Deci-AI/super-gradients)
- Train custom models
- No SSL expertise required
- Automatic SSL method selection (soon!)
- Python, Command Line, and Docker support
- Multi-GPU and multi-node support
- Export models for fine-tuning or inference
- Generate and export image embeddings
- Monitor training progress with TensorBoard, Weights & Biases, Neptune, etc. (soon!)

## License

Lightly**Train** is available under an AGPL-3.0 and a commercial license. Please contact us
at info@lightly.ai for more information.

## Contact

- [**Email**](info@lightly.ai)
- [**Website**](https://www.lightly.ai/lightlytrain)
- [**Discord**](https://discord.gg/xvNJW94)
