基于知识蒸馏的模型训练方法、装置、电子设备、计算机可读存储介质及计算机程序产品

本申请提供了一种基于知识蒸馏的模型训练方法、装置、电子设备、计算机可读存储介质及计算机程序产品；方法包括：获取第一样本数据集；第一样本数据集包括多个第一标注文本，每一第一标注文本对应的第一样本图像具有不同视觉效果类型且包含多个对象；调用预训练的教师模型，基于第一标注文本生成第一生成图像；通过待训练的学生模型，基于第一生成图像和第一标注文本生成第二生成图像；基于第一生成图像和第二生成图像，确定学生模型的知识蒸馏损失；基于知识蒸馏损失，对学生模型进行模型参数更新，得到训练后的学生模型。通过本申请，不仅能够通过训练后的学生模型生成具有不同视觉效果且包含多个对象的图像，还能够提升图像的生成质量。