search

lucidrains / voicebox-pytorch

Implementation of Voicebox, new SOTA Text-to-speech network from MetaAI, in Pytorch
MIT License
589 stars 49 forks source link

Loss calculation #34

Closed YKoustubhRao closed 10 months ago

YKoustubhRao commented 10 months ago

I think, cond_mask_with_pad_dim should also be applied to target (flow) while calculating loss.

lucidrains commented 10 months ago

@YKoustubhRao it is here cond_mask_with_pad_dim is derived from cond_mask