模型训练方法及其装置

本申请公开了一种模型训练方法及其装置，属于信息技术领域。该方法包括：对第一图像数据集按图像数据尺寸进行归类处理，得到N类图像数据，同一类图像数据的尺寸相同；根据N类图像数据中每类图像数据的尺寸，确定每类图像数据对应的批次大小，不同类图像数据对应的批次大小不同；分别按照每类图像数据对应的批次大小，将每类图像数据分为多个批次图像数据，得到由N类图像数据中的各个批次图像数据构成的训练数据集；利用训练数据集对图像处理模型进行多轮迭代训练，每轮训练使用训练数据集中的一个批次图像数据。