Overview
A complete repository for fine-tuning BERT and similar encoder models. Includes data loaders, training loops with mixed precision, evaluation metrics, and export to ONNX. Suitable for classification, NER, and QA. Comes with example notebooks and Docker setup.
What's included
- Training & eval scripts
- Docker & configs
- Example notebooks
- ONNX export
Requirements
- Python 3.9+
- CUDA-capable GPU recommended
Repository
Private GitHub repository. Access is emailed after purchase — create a marketplace account at checkout.
Related products
Repository
Vision Model Inference API
Fast inference server for vision models (CLIP, ViT) with batching and REST API.
$99View details
RepositorySST ASR Fine-Tuning Kit
Speech-to-text fine-tuning for wav2vec2/Whisper-style models with SST datasets.
$189View details
RepositoryMLOps Pipeline Starter
CI/CD and training pipeline for ML models with experiment tracking and deployment.
$249View details