mlcommons / training

Reference implementations of MLPerf™ training benchmarks
https://mlcommons.org/en/groups/training
Apache License 2.0
1.57k stars 548 forks source link

Add quick SSD demo #683

Closed davidjurado closed 3 months ago

davidjurado commented 8 months ago

This PR includes a small demo to execute the SSD benchmark.

Project setup

# Create Python environment and install MLCube Docker runner 
virtualenv -p python3 ./env && source ./env/bin/activate && pip install mlcube-docker

# Fetch the implementation from GitHub
git clone https://github.com/mlcommons/training && cd ./training/single_stage_detector
git fetch origin pull/683/head:feature/ssd_example && git checkout feature/ssd_example

Go to mlcube directory and study what tasks MLCube implements.

cd ./mlcube
mlcube describe

MLCube tasks

Download demo.

mlcube run --task=download_demo -Pdocker.build_strategy=always

Download model.

mlcube run --task=download_model -Pdocker.build_strategy=always

Run demo.

mlcube run --task=demo -Pdocker.build_strategy=always

Execute the complete pipeline

You can execute the complete pipeline with one single command.

mlcube run --task=download_demo,download_model,demo -Pdocker.build_strategy=always
github-actions[bot] commented 8 months ago

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

ahmadki commented 8 months ago

LGTM