pytorch-trainer

Overview

PyTorch model training skill with custom training loops, gradient management, GPU optimization, and integration with experiment tracking systems.

Capabilities

• Custom training loop execution
• Learning rate scheduling (StepLR, CosineAnnealing, OneCycleLR, etc.)
• Gradient clipping and accumulation
• Mixed precision training (AMP)
• Checkpoint management and resumption
• DataLoader optimization
• Multi-GPU training (DataParallel, DistributedDataParallel)
• Early stopping with patience

Target Processes