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
26 watching
360 forks
Language: Python
last commit: 3 months ago
Linked from 1 awesome list
aicomputer-visiondeep-learningmachine-learningpythonpytorch
krzjoa/awesome-python-data-scienceRelated 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. | 897 |
 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. | 40 |
 tristandeleu/pytorch-maml-rl | Replication of Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks in PyTorch for reinforcement learning tasks | 830 |
 graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 572 |
 zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,044 |
 eduardoleao052/js-pytorch | A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. | 1,093 |
 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,078 |
 uber/petastorm | Enables training and evaluation of deep learning models from Apache Parquet datasets in various machine learning frameworks | 1,805 |
 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 | 679 |
| open-mmlab/mmgeneration | Generative model toolkit for GANs and image generation tasks | 1,925 |
 deepmimo/deepmimo-matlab | Provides MATLAB code and dataset for training machine learning models in millimeter wave and massive MIMO systems | 162 |