用于生成代码的神经网络模型的训练方法和基于神经网络模型生成代码的方法和装置
申请号:CN202411978111
申请日期:2024-12-30
公开号:CN119938024A
公开日期:2025-05-06
类型:发明专利
摘要
提供了一种用于生成代码的神经网络模型的训练方法、基于神经网络模型生成代码的方法和装置。训练方法包括主训练阶段,主训练阶段包括:获取第一图像,其中,第一图像包含第一页面布局;标注第一图像的第一标签值,其中,第一标签值用于表示生成第一页面布局的代码;将第一图像输入到视觉编码器,获取视觉编码器的所输出的第一图像的第一特征向量;将第一特征向量输入到适配器,获取适配器的所输出的第二特征向量;将第二特征向量输入到语言模型,获取语言模型所输出的用于生成第一页面布局的第一代码表示;基于第一代码表示和第一标签值,计算第一损失值;以及基于第一损失值,调整视觉编码器、适配器和语言模型的参数。
技术关键词
神经网络模型
适配器
生成代码
布局
页面
标签
词典
阶段
索引
模块
处理器
训练装置
参数
图像块
视频编码器
计算机程序产品
指令
计算机设备
可读存储介质