MindPet 微调算法套件开源项目

我要开发同款
白开水不加糖2023年03月29日
61阅读
开发技术Python
所属分类人工智能
授权协议Apache-2.0

作品详情

MindPet(Pet:Parameter-EfficientTuning)是属于Mindspore领域的微调算法套件。随着计算算力不断增加,大模型无限的潜力也被挖掘出来。但随之在应用和训练上带来了巨大的花销,导致商业落地困难。因此,出现一种新的参数高效(parameter-efficient)算法,与标准的全参数微调相比,这些算法仅需要微调小部分参数,可以大大降低计算和存储成本,同时可媲美全参微调的性能。

微调算法API目前MindPet已提供以下五种经典低参微调算法以及一种提升精度的微调算法的API接口,用户可快速适配原始大模型,提升下游任务微调性能和精度;

微调算法算法论文使用说明LoRALoRA:Low-RankAdaptationofLargeLanguageModelsTK_DeltaAlgorithm_README 第一章PrefixTuningPrefix-Tuning:OptimizingContinuousPromptsforGenerationTK_DeltaAlgorithm_README 第二章AdapterParameter-EfficientTransferLearningforNLPTK_DeltaAlgorithm_README 第三章LowRankAdapterCompacter:Efficientlow-rankhypercomplexadapterlayersTK_DeltaAlgorithm_README 第四章BitFitBitFit:SimpleParameter-efficientFine-tuningforTransformer-basedMaskedLanguage-modelsTK_DeltaAlgorithm_README 第五章R_DropR-Drop:RegularizedDropoutforNeuralNetworksTK_DeltaAlgorithm_README 第六章共性图操作API冻结指定模块功能APIMindPet支持用户根据微调算法或模块名冻结网络中部分模块,提供调用接口和配置文件两种实现方式。

使用说明参考TK_GraphOperation_README 第一章。

保存可训练参数功能APIMindPet支持用户单独保存训练中可更新的参数为ckpt文件,从而节省存储所用的物理资源。

使用说明参考TK_GraphOperation_README 第二章。

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论