最后，读出头既可以进行训练，也可以使用。我们可以在单张Nvidia A100 40GB GPU上，用100个样本进行3小时的训练，为Stable Diffusion XL训练一个读出头（5.9M参数）。