PyPI page
Home page
Author:
None
Summary:
Config-driven PyTorch training framework: assemble nets, data, losses and trainers from YAML
Latest version:
0.7.0
Required dependencies:
albumentations
|
colorama
|
hydra-core
|
numpy
|
omegaconf
|
opencv-python
|
pandas
|
scikit-learn
|
scipy
|
tensorboard
|
timm
|
torch
|
torchmetrics
|
torchvision
|
tqdm
Optional dependencies:
faster-coco-eval
|
mkdocs
|
mkdocs-material
|
mosaicml
|
onnx
|
onnxruntime
|
segmentation-models-pytorch
Downloads last day:
321
Downloads last week:
694
Downloads last month:
694