The dataset of coco_traint2i

[xinding64]

could you please provide more details about MS-COCO training,especially where to download coco's image and prompt .we couldn't find the prompt.txt you used

[yhy-2000]

According to my understanding, 'prompt.txt' isn't used when calculating FID score, but used in generate a demo grid image.

U-ViT/train_t2i_discrete.py

Line 246 in ce55170

contexts = torch.tensor(dataset.contexts, device=device)[: 2 * 5]