PyPI page
Home page
Author:
Shabanov Aleksei
License:
Apache License 2.0
Summary:
OML is a PyTorch-based framework to train and validate the models producing high-quality embeddings.
Latest version:
4.0.0
Required dependencies:
albumentations
|
gdown
|
matplotlib
|
numpy
|
opencv-python-headless
|
pandas
|
pillow
|
tensorboard
|
torch
|
torchvision
|
validators
Optional dependencies:
clearml
|
grad-cam
|
hydra-core
|
ipython
|
mlflow
|
neptune
|
pre-commit
|
pyright
|
pytest
|
python-dotenv
|
pytorch-lightning
|
soundfile
|
torchaudio
|
transformers
|
wandb
Downloads last day:
47
Downloads last week:
283
Downloads last month:
1,290