ClassyVision
An end-to-end PyTorch framework for image and video classification
...Unlike traditional computer vision libraries that focus solely on modular components, Classy Vision provides a complete and unified framework, featuring distributed training, reproducible experiments, and flexible configuration tools. It offers high performance and scalability—capable of training models like ResNet-50 on ImageNet in just minutes—while remaining accessible to both researchers and production engineers. The library integrates seamlessly with PyTorch Hub for easy access to pretrained models and supports elastic training using PyTorch Elastic, making distributed training robust to changes in cluster resources or hardware failures.