img_cat_cond referenced before assignment error #2
Description
else: ## simply repeat the cond_frame to match the seq_len of z # img_cat_cond = z_cat[:,:,cond_frame_index,:,:] img_cat_cond = img_cat_cond.unsqueeze(2) # not defined before this point img_cat_cond = repeat(img_cat_cond, 'b c t h w -> b c (repeat t) h w', repeat=z.shape[2]) Here is the error message:
Mask2IV/main/../lvdm/models/ddpm3d_maskgen.py", line 1158, in get_batch_input img_cat_cond = img_cat_cond.unsqueeze(2) UnboundLocalError: local variable 'img_cat_cond' referenced before assignment Going to try to figure out what img_cat_cond is supposed to be but would be helpful if this got fixed. Thanks!