diogocarapito/hf-mlflow-template

Template for a Hugging Face fine-tuning classification model, pytorch based with local MLFlow server tracking

Python

0

6 commits

updated Aug 11, 2024

See the code

README

hf-mlflow-template

Template for a Hugging Face fine-tuning classification model, pytorch based with local MLFlow server tracking

Github Actions Workflow

cheat sheet

venv

create venv

python3 -m venv .venv

activate venv

source .venv/bin/activate

Docker

build docker image

docker build -t main:latest .