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

GitHub

1k stars
25 watching
355 forks
Language: Python
last commit: 15 days ago
Linked from 1 awesome list

aicomputer-visiondeep-learningmachine-learningpythonpytorch

Backlinks from these awesome lists:

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