T2I_CL icon indicating copy to clipboard operation
T2I_CL copied to clipboard

add: mask_correlated_samples_2 in AttnGAN+CL

Open JoonHong-Kim opened this issue 3 years ago • 2 comments

I think the pretrain_DAMSM.py works reliably only when this function is also at masks.py in the AttnGAN+CL folder.

JoonHong-Kim avatar Dec 08 '21 11:12 JoonHong-Kim

Maybe. My experiments for the pretrain_DAMSM.py were conducted under the AttnGAN+CL.

huiyegit avatar Dec 08 '21 14:12 huiyegit

The mask_correlated_samples_2 function is used in AttnGAN+CL/code/pretrain_DAMSM.py, but it was not defined in the masks.py file in the same folder, so I modified it. In this case, DM-GAN+CL/code/masks.py outside file is imported, there is no problem functionally, but if the two folders are separated, there may be a problem.

JoonHong-Kim avatar Dec 08 '21 14:12 JoonHong-Kim