forked from NVIDIA/NeMo
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[TTS] Fixing RADTTS training - removing view buffer and fixing accura…
…cy issue (NVIDIA#5358) * Fixing RADTTS training - removing view buffer and fixing accuracy issue Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Addressing code review Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Addressing code review 2 Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixed assignment Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Working script Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * restored flatten_parameters Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Working bias alias for export Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Removing unused import Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Reverting PartialConv Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Removing flatten_parameters Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Moving mask updater to GPU Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Restored norms Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Restored flatten Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Moved to sort/unsort Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Moved to masked norm Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Turned off cache Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * cleanup Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Verifying cache not used Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Removing cache Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Working autocast export Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * restored e-6 Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Removed some casts around masks, etc Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixing some casts Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixing in-place ops Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * fixing grad Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Small export fixes Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * LGTM cleanup Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixed lstm_tensor Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * restored TS check routine Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixed config error Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * reverting some bad optimizations Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * [TTS] add CI test for RADTTS training recipe. Signed-off-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com> * Addressing code review Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Removing unused var Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Adding debug Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Logging fixes Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixing training warnings Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixing more warnings Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Fixing more warnings 2 Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Code review fixes Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Improving TS check Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Addressing code review comments, optimizing script Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Forced no-autocast Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> Signed-off-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com> Co-authored-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com> Co-authored-by: Oleksii Kuchaiev <okuchaiev@users.noreply.github.com> Signed-off-by: Hainan Xu <hainanx@nvidia.com>
- Loading branch information
1 parent
ea61803
commit 4125ab7
Showing
12 changed files
with
240 additions
and
254 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.