facebookresearch/pycls
A PyTorch research codebase for image classification implementing standard deep learning models with multi-GPU training support.

Velocity · 7d
+0.8
★ / day
Trend
→steady
star history
pycls is an image classification research framework built in PyTorch, originally developed for visual recognition design space research. It supports rapid implementation and evaluation of research ideas with implementations of standard vision architectures including ResNet, ResNeXt, EfficientNet, and RegNet. The codebase includes a model zoo with pretrained weights and baseline results across various computational regimes.