Problem with AttnGAN training

[ecfish]

Thanks for this excellent work! I have some problem with AttnGAN training. I am not sure whether i have a correct thought of the following code.

/code/trainer.py
line 246 ~ 271
imgs, captions, cap_lens, class_ids, keys = prepare_data(data)
errD = discriminator_loss(netsD[i], imgs[i], fake_imgs[i],
                                              sent_emb, real_labels, fake_labels)

I think netsD[i] is the ith discriminator used in AttnGAN. In paper, there are three discriminators. I also think that imgs are corresponding to captions and have a size equal to batch size. So, how can parameter i represents both the netsD index and img index.

[ecfish]

sorry, my fault