Model Training · seb1n/awesome-ai-agent-skills

Train a machine learning model end-to-end

Handles the full training lifecycle — data loading, preprocessing, architecture selection, training loops with validation, learning-rate scheduling, and checkpointing — across PyTorch, TensorFlow, and scikit-learn. Useful when setting up or improving a model training pipeline.

Good for

  • Set up a training pipeline for a new model
  • Add validation and early stopping to a training loop
  • Export a trained checkpoint for deployment
Category
Coding

Open-source skills are maintained by their authors and listed as published, with attribution. Results depend on how well the skill fits your task and material.

A good place to start

Train an image classifier on my labeled dataset, validate each epoch, and save the best checkpoint.

Make your next great thing.

Bring a question, a file, or an idea that’s not quite there yet.