mmpretrain
MMPreTrain是基于PyTorch的开源预训练工具箱,提供丰富的训练策略和高效的模型分析工具。支持多种主干网络和预训练模型,如VGG、ResNet、Vision-Transformer等。具备强大的扩展性和高效性,适用于图像分类、图像描述、视觉问答等多种推理任务。最新版本v1.2.0增加了对LLaVA 1.5和RAM的支持,并提供Gradio界面。适用于多模态学习和自监督学习,支持大规模训练配置。提供详细的安装和教程文档,帮助用户快速入门。