GitXplorerGitXplorer
J

mlflow_training_pipeline

public
1 stars
1 forks
0 issues

Commits

List of commits on branch main.
Unverified
19e46f1ba20b6de103c0679d51e9a4f1a5884866

revised readme

JJINO-ROHIT committed a year ago
Unverified
c323e4c2127ea8cf0201d9f7ed73eb4ead92fd36

finishing up

JJINO-ROHIT committed a year ago
Verified
6e516220cf3a68b9d65edeb2f0464d25c685d2e4

Update README.md

JJINO-ROHIT committed a year ago
Verified
ca7210ddc08acf01ef1d74755df5d62f76fb11d6

Initial commit

JJINO-ROHIT committed a year ago

README

The README file for this repository.

Mlflow Pytorch Pipeline

This is a simple training pipeline for a segmentation model(Unet + Effnet) implemented in pytorch along with mlflow tracking.

Installation

  1. Create a virtual environment
python3 -m venv hubmap
source hubmap/bin/activate
  1. Install the requirements
make dev
  1. Begin the training
make train
  1. Launch the UI to see the tracked parameters and models
make ui