# LightlyTrain
*Train models with self-supervised learning in a single command*
## Why LightlyTrain
LightlyTrain 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 LightlyTrain result in improved performance, faster convergence, and
better generalization compared to models trained without SSL. Image embeddings created
with LightlyTrain 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 LightlyTrain 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), 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 (soon!) support
* Export models for fine-tuning or inference
* Generate and export image embeddings
* Monitor training progress with TensorBoard, Weights & Biases, Neptune, etc. (soon!)
## License
LightlyTrain 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)
Raw data
{
"_id": null,
"home_page": null,
"name": "lightly-train",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.8",
"maintainer_email": null,
"keywords": "machine-learning, computer-vision, deep-learning, self-supervised-learning, contrastive-learning, pytorch, python, pretrained-models, embeddings",
"author": "Lightly Team",
"author_email": null,
"download_url": "https://files.pythonhosted.org/packages/6a/df/242102a4bac0217d0c7a33d446181e9972668af07aa3a1d525e3dd4002ba/lightly_train-0.2.5.tar.gz",
"platform": null,
"description": "# LightlyTrain\n\n*Train models with self-supervised learning in a single command*\n\n\n## Why LightlyTrain\n\nLightlyTrain uses self-supervised learning (SSL) to train models on large datasets\nwithout the need for labels. It provides simple Python, Command Line, and Docker\ninterfaces to train models with popular SSL methods such as SimCLR or DINO. The trained\nmodels are ideal starting points for fine-tuning on downstream tasks such as image \nclassification, object detection, and segmentation or for generating image embeddings.\nModels trained with LightlyTrain result in improved performance, faster convergence, and\nbetter generalization compared to models trained without SSL. Image embeddings created\nwith LightlyTrain capture more relevant information than their supervised counterparts\nand seamlessly extend to new classes due to the unsupervised nature of SSL.\n\nLightly is the expert in SSL for computer vision and developed LightlyTrain to simplify\nmodel training for any task and dataset.\n\n\n## Features\n\n* Train models on any image data without labels\n* Train models from popular libraries such as [torchvision](https://github.com/pytorch/vision), [TIMM](https://github.com/huggingface/pytorch-image-models), and [SuperGradients](https://github.com/Deci-AI/super-gradients)\n* Train custom models\n* No SSL expertise required\n* Automatic SSL method selection (soon!)\n* Python, Command Line, and Docker support\n* Multi-GPU and multi-node (soon!) support\n* Export models for fine-tuning or inference\n* Generate and export image embeddings\n* Monitor training progress with TensorBoard, Weights & Biases, Neptune, etc. (soon!)\n\n## License\n\nLightlyTrain is available under an AGPL-3.0 and a commercial license. Please contact us\nat info@lightly.ai for more information.\n\n## Contact\n\n* [**Email**](info@lightly.ai) \n* [**Website**](https://www.lightly.ai/lightlytrain) \n* [**Discord**](https://discord.gg/xvNJW94)\n",
"bugtrack_url": null,
"license": "AGPL-3.0",
"summary": "Train models with self-supervised learning in a single command",
"version": "0.2.5",
"project_urls": null,
"split_keywords": [
"machine-learning",
" computer-vision",
" deep-learning",
" self-supervised-learning",
" contrastive-learning",
" pytorch",
" python",
" pretrained-models",
" embeddings"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "16d876d1129116fa2de3dfd58c2e30a6e73ef03cfc46550b6763f37fcfd456b8",
"md5": "90757178aa028896082af637237bd291",
"sha256": "aa25bbecf8c388f38fc9c3b696df600cfb93fb8f2c353a5036d969eb1f56f253"
},
"downloads": -1,
"filename": "lightly_train-0.2.5-py3-none-any.whl",
"has_sig": false,
"md5_digest": "90757178aa028896082af637237bd291",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.8",
"size": 98277,
"upload_time": "2024-09-24T14:12:20",
"upload_time_iso_8601": "2024-09-24T14:12:20.321285Z",
"url": "https://files.pythonhosted.org/packages/16/d8/76d1129116fa2de3dfd58c2e30a6e73ef03cfc46550b6763f37fcfd456b8/lightly_train-0.2.5-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "6adf242102a4bac0217d0c7a33d446181e9972668af07aa3a1d525e3dd4002ba",
"md5": "c3dc6ae44789df24cd19601975821b37",
"sha256": "3bc3de6917e60e57f9f6dca761744e76b09696abd95699fc41beba1be8a0af9a"
},
"downloads": -1,
"filename": "lightly_train-0.2.5.tar.gz",
"has_sig": false,
"md5_digest": "c3dc6ae44789df24cd19601975821b37",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.8",
"size": 91898,
"upload_time": "2024-09-24T14:12:23",
"upload_time_iso_8601": "2024-09-24T14:12:23.161413Z",
"url": "https://files.pythonhosted.org/packages/6a/df/242102a4bac0217d0c7a33d446181e9972668af07aa3a1d525e3dd4002ba/lightly_train-0.2.5.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-09-24 14:12:23",
"github": false,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"lcname": "lightly-train"
}
JSON