Loss calculation
YKoustubhRao opened this issue · 1 comments
YKoustubhRao commented
I think, cond_mask_with_pad_dim should also be applied to target (flow) while calculating loss.
lucidrains commented
@YKoustubhRao it is here cond_mask_with_pad_dim is derived from cond_mask