/mlflow_pytorch_exp

mlflow-pytorch-exp helps create a boilerplate for an mlflow pytorch project in a single line in the terminal

Primary LanguagePythonMIT LicenseMIT

https://img.shields.io/badge/PyTorch-EE4C2C?style=for-the-badge&logo=pytorch&logoColor=white https://img.shields.io/badge/MLflow-0194E2.svg?style=for-the-badge&logo=MLflow&logoColor=white

mlflow_pytorch_exp

install from PyPI

pip install mlflow-pytorch-exp==0.0.1

Create a project

mlflow-pytorch-exp create <path/to/folder>

Insall manually

  1. Build the packege:
python setup.py sdist bdist_wheel
  1. Install package :
pip install dist/mlflow_pytorch_exp-0.0.1-py3-none-any.whl
  1. Create project
mlflow-pytorch-exp create <path/to/folder>

example :

mlflow-pytorch-exp create my_project
mlflow_pytorch_exp
├── LICENSE
├── MANIFEST.in
├── mlflow_pytorch_exp
│   ├── __init__.py
│   ├── mlflow_pytorch_exp.py
│   └── templates
│       ├── checkpoints
│       ├── config
│       │   └── config.py
│       ├── data
│       ├── dataloader
│       │   └── dataset.py
│       ├── losses
│       │   └── loss.py
│       ├── metrics
│       │   └── metric.py
│       ├── models
│       │   └── model.py
│       ├── train.py
│       └── utils
│           └── utils.py
├── README.md
└── setup.py