{"repo":"open-mmlab/mmengine","free":true,"listed":false,"github":"https://github.com/open-mmlab/mmengine","clone":"git clone https://github.com/open-mmlab/mmengine.git","description":"OpenMMLab Foundational Library for Training Deep Learning Models","language":"Python","stars":1488,"topics":["computer-vision","deep-learning","pytorch","ai","machine-learning","python"],"license":"Apache-2.0","category":"machine-learning","readme_excerpt":"  OpenMMLab website HOT      OpenMMLab platform TRY IT OUT   Introduction Installation Get Started 📘Documentation 🤔Reporting Issues English 简体中文 What's New v0.10.6 was released on 2025-01-13. Highlights: - Support custom artifact location in MLflowVisBackend #1505 - Enable exclude frozen parameters for DeepSpeedEngine. zero3 consolidated 16bit state dict #1517 Read Changelog for more details. Introduction MMEngine is a foundational library for training deep learning models based on PyTorch. It serves as the training engine of all OpenMMLab codebases, which support hundreds of algorithms in various research areas. Moreover, MMEngine is also generic to be applied to non-OpenMMLab projects. Its highlights are as follows: Integrate mainstream large-scale model training frameworks - ColossalAI - DeepSpeed - FSDP Supports a variety of training strategies - Mixed Precision Training - Gradient Accumulation - Gradient Checkpointing Provides a user-friendly configuration system - Pure Python-style configuration files, easy to navigate - Plain-text-style configuration files, supporting JSON and YAML Covers mainstream training monitoring platforms - TensorBoard WandB MLflow - ClearML Neptune DVCLive Aim Installation Supported PyTorch Versions MMEngine PyTorch Python ------------------ ------------ -------------- main =1.6 \\ =3.8, \\ =0.9.0, \\ =1.6 \\ =3.8, \\ Before installing MMEngine, please ensure that PyTorch has been successfully installed following the ","default_branch":null,"files":null,"tree":[],"storefront":"/r/open-mmlab","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/open-mmlab/mmengine/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}