Vision Tasks
Train models for image classification, object detection, and vision-language tasks.Image Classification
Quick Start
Parameters
Data Format
Folder structure:Example: Custom Classifier
Image Regression
For predicting continuous values from images.Quick Start
Parameters
Example: Age Prediction
Object Detection
Quick Start
Data Format
COCO-style format:Parameters
Example: Custom Detection
Vision-Language Models (VLM)
Train models that understand both images and text.Quick Start
Data Format
Parameters
Example: Image Captioning
Common Models
Image Classification
Object Detection
Vision-Language
GPU Memory Tips
- Use smaller batch sizes for large images
- Enable gradient checkpointing for VLMs
- Use LoRA for VLM training:
Next Steps
Text Tasks
Text classification and NER
Tabular Data
Train on structured data