flux 辨别器forward  bug

作者您好，https://github.com/RedAIGC/Target-Driven-Distillation/blob/9551bddf4cf501948927b2890f87d1c4334f391d/train/FLUX/pcm_discriminator_flux.py#L236C13-L236C16 我认为这里可能是个bug。这里本来意图可能是对所有辨别头，要所有层的feature都依次流过，但现在的实现是同时迭代每一层内每个batch的每个样本，在独立的过一个对应的检测器头，这会导致每个检测头只检测到某一个样本子集。