mmengine
Deep learning trainer
Provides a flexible and configurable framework for training deep learning models with PyTorch.
OpenMMLab Foundational Library for Training Deep Learning Models
1k stars
25 watching
355 forks
Language: Python
last commit: 15 days ago
Linked from 1 awesome list
aicomputer-visiondeep-learningmachine-learningpythonpytorch
Related projects:
Repository | Description | Stars |
---|---|---|
oml-team/open-metric-learning | A PyTorch-based framework for training and validating models that produce high-quality embeddings for computer vision and other tasks. | 882 |
openbmb/cpm-live | A live training platform for large-scale deep learning models, allowing community participation and collaboration in model development and deployment. | 511 |
dmarnerides/pydlt | A PyTorch-based toolbox for building and training deep learning models with ease. | 204 |
erotemic/netharn | A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. | 39 |
tristandeleu/pytorch-maml-rl | Replication of Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks in PyTorch for reinforcement learning tasks | 827 |
graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 569 |
zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,041 |
eduardoleao052/js-pytorch | A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. | 1,084 |
wyy-123-xyy/ra-fed | A Python implementation of a distributed machine learning framework for training neural networks on multiple GPUs | 6 |
learning-at-home/hivemind | A PyTorch library for decentralized deep learning across the Internet. | 2,051 |
uber/petastorm | Enables training and evaluation of deep learning models from Apache Parquet datasets in various machine learning frameworks | 1,799 |
ardanlabs/training-ai | Provides training materials and tools for building machine learning applications | 72 |
volcengine/vescale | A PyTorch-based framework for training large language models in parallel on multiple devices | 663 |
open-mmlab/mmgeneration | Generative model toolkit for GANs and image generation tasks | 1,909 |
deepmimo/deepmimo-matlab | Provides MATLAB code and dataset for training machine learning models in millimeter wave and massive MIMO systems | 156 |