search

lucidrains / spear-tts-pytorch

Implementation of Spear-TTS - multi-speaker text-to-speech attention network, in Pytorch
MIT License
249 stars 18 forks source link

Minor trainer cleanup #7

Closed lucasnewman closed 11 months ago

lucasnewman commented 11 months ago

This prevents a possible issue that prevents training when using the new mask_id parameter, cleans up some unused parameters in the trainer that could be misleading, and handles the logits properly from the wrappers.