Open JJJYmmm opened 10 months ago
Actually, the origin code won't get in the branch. Because this time vae is ddp model. Correct code should be below.
if hasattr(vae.module, 'get_codebook_indices'):
code = vae.module.get_codebook_indices(mask)
remask = vae.module.decode(code)[0, 0, :, :].cpu().numpy()
Hello! I find that
/vae/utils/eval_coco.py
denorm the reconstruction image twice in line 45.because in class func
decode
, the attruse_norm
is True, sodecode
will denorm the image, but the code denorm after decodeing. I will try to investigate the effect when evaluating.