ycai4591679 / Noah-Gift--mlops-template-DeepLearnin

mlops template

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CI Codespaces Prebuilds

Template for MLOPs projects with GPU

  1. First thing to do on launch is to open a new shell and verify virtualenv is sourced.

Things included are:

  • Makefile

  • Pytest

  • pandas

  • Pylint

  • Dockerfile

  • GitHub copilot

  • jupyter and ipython

  • Most common Python libraries for ML/DL and Hugging Face

  • githubactions

Two fun tools to explore:

  • Zero-shot classification: ./hugging-face/zero_shot_classification.py classify
  • Yake for candidate label creation: ./utils/kw_extract.py

Try out Bento

docker run -it --rm -p 8888:8888 -p 3000:3000 -p 3001:3001 bentoml/quickstart:latest

Verify GPU works

The following examples test out the GPU

  • run pytorch training test: python utils/quickstart_pytorch.py
  • run pytorch CUDA test: python utils/verify_cuda_pytorch.py
  • run tensorflow training test: python utils/quickstart_tf2.py
  • run nvidia monitoring test: nvidia-smi -l 1 it should show a GPU
  • run whisper transcribe test ./utils/transcribe-whisper.sh and verify GPU is working with nvidia-smi -l 1

Additionally, this workspace is setup to fine-tune Hugging Face

fine-tune

python hf_fine_tune_hello_world.py

Used in Following Projects

Used as the base and customized in the following Duke MLOps and Applied Data Engineering Coursera Labs:

References

About

mlops template

License:MIT License


Languages

Language:Python 89.4%Language:Makefile 4.6%Language:Dockerfile 3.8%Language:Shell 2.2%