带有标签batch generation的模型