模型训练方法、装置、计算机设备及存储介质

本公开涉及计算机技术领域，公开了模型训练方法、装置、计算机设备及存储介质，该方法包括：在对目标分子生成模型的预训练期间，对用于训练的分子结构进行片段切分，得到用于训练的分子结构的多个片段；从该多个片段中选择用于训练的分子结构的目标片段集合，该目标片段集合为多个片段的一部分；获取目标片段集合的特征；生成用于训练的分子结构的整体特征，用于训练的分子结构的整体特征包括：用于训练的分子结构中每个原子的局部电荷；利用目标分子生成模型，根据目标片段集合的特征、用于训练的分子结构的整体特征，生成目标片段集合对应的预测分子结构，以及生成目标片段集合对应的预测分子结构与用于训练的分子结构之间的损失。