mmengine
MMEngine is a foundational library facilitating PyTorch-based deep learning model training, integrating popular frameworks such as ColossalAI and DeepSpeed. It offers various training strategies, a flexible configuration system, and supports multiple monitoring platforms including TensorBoard and MLflow. Serving as the training core for OpenMMLab projects, MMEngine is adaptable for diverse research areas and wider applications. The latest updates include customizable MLflow artifact locations and enhanced DeepSpeed parameter control.
