mirror of
https://github.com/wassname/TTS.git
synced 2026-10-28 12:41:20 +08:00
BCE masked loss and padding stop_tokens with 0s not 1s
This commit is contained in:
1 parent
a5d919cec8
commit
a68012aec2
3 files changed
+34
-5
No files matched your search
+1
-1
@@ -31,7 +31,7 @@ def prepare_tensor(inputs, out_steps):
|
||||
|
||||
|
||||
def _pad_stop_target(x, length):
|
||||
_pad = 1.
|
||||
_pad = 0.
|
||||
assert x.ndim == 1
|
||||
return np.pad(
|
||||
x, (0, length - x.shape[0]), mode='constant', constant_values=_pad)
|
||||
|
||||
Reference in new issue
Block a user