Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

MLCube implementation for ResNet #686

Open
wants to merge 10 commits into
base: master
Choose a base branch
from

Conversation

davidjurado
Copy link
Contributor

@davidjurado davidjurado commented Oct 19, 2023

This PR includes the MLCube implementation for the image classification benchmark with Resnet, it also includes a small demo dataset to easily reproduce the benchmark.

Project setup

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

# Fetch the implementation from GitHub
git clone https://github.com/mlcommons/training && cd training
git fetch origin pull/686/head:feature/resnet_mlcube && git checkout feature/resnet_mlcube
cd image_classification/mlcube

Go to mlcube directory and study what tasks MLCube implements.

cd ./mlcube
mlcube describe

MLCube tasks

For the entire IMAGENET dataset, you will need to download the complete dataset and place it in the workspace under the mlcube folder, then you can use the following tasks:

Process dataset.

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

Train RESNET.

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

Run compliance checker.

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

Running a small demo

To download the susample dataset and run the demo use the following command:

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

@davidjurado davidjurado requested a review from a team as a code owner October 19, 2023 22:43
@github-actions
Copy link

github-actions bot commented Oct 19, 2023

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

@davidjurado davidjurado changed the title MLCube implementation for Resnet MLCube implementation for ResNet Oct 19, 2023
@davidjurado davidjurado changed the title MLCube implementation for ResNet [WIP] MLCube implementation for ResNet Oct 19, 2023
@davidjurado davidjurado changed the title [WIP] MLCube implementation for ResNet MLCube implementation for ResNet Feb 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant