Train models with self-supervised learning in a single command
Project description
LightlyTrain
Train models with self-supervised learning in a single command
Installation
git clone https://github.com/lightly-ai/lightly-train.git
pip install lightly-train[tensorboard,timm]
Usage
Python
Training
import lightly_train
lightly_train.train(
out="my_output_dir",
data="my_data_dir",
model="torchvision/resnet18",
method="simclr",
batch_size=128,
epochs=100,
precision="16-mixed",
optim_args=dict(lr=0.0001),
method_args=dict(temperature=0.1),
)
In most cases you only have to specify out
, data
, and model
. The rest is optional.
The training process can be monitored with TensorBoard (requires
pip install lightly-train[tensorboard]
):
tensorboard --logdir my_output_dir
Exporting
import lightly_train
lightly_train.export(
out="my_output_dir/model_state_dict.pt",
checkpoint="my_output_dir/checkpoints/last.ckpt",
part="model",
format="torch_state_dict",
)
Embedding
import lightly_train
lightly_train.embed(
out="my_output_dir/embeddings.csv",
data="my_data_dir",
checkpoint="my_output_dir/checkpoints/last.ckpt",
format="csv",
)
Supported Models
import lightly_train
print(lightly_train.list_models())
Supported Methods
import lightly_train
print(lightly_train.list_methods())
Command Line
Help
lightly-train help
Training
lightly-train train \
out=my_output_dir \
data=my_data_dir \
model=torchvision/resnet18 \
method=simclr \
batch_size=128 \
epochs=100 \
precision=16-mixed \
optim_args.lr=0.0001 \
method_args.temperature=0.1
In most cases you only have to specify out
, data
, and model
. The rest is optional.
The training process can be monitored with TensorBoard (requires
pip install lightly-train[tensorboard]
):
tensorboard --logdir my_output_dir
Embedding
lightly-train embed \
out=my_output_dir/embeddings.csv \
data=my_data_dir \
checkpoint=my_output_dir/checkpoints/last.ckpt \
format=csv
Exporting
lightly-train export \
out=my_output_dir/model_state_dict.pt \
checkpoint=my_output_dir/checkpoints/last.ckpt \
part=model \
format=torch_state_dict
Supported Models
lightly-train list_models
Supported Methods
lightly-train list_methods
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
lightly_train-0.2.1.tar.gz
(71.2 kB
view hashes)
Built Distribution
Close
Hashes for lightly_train-0.2.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bb724a0d277650180721139903ad5e0a6d57f1fba6d4a0a16f998fdbb75bae4a |
|
MD5 | cfa2eed13b407847e489a8569da62721 |
|
BLAKE2b-256 | a4588a75cb2d50414b545b023e234bebb6c9e2619b08f94811e9862ecd7f1f49 |