mirror of
https://github.com/wassname/TTS.git
synced 2026-09-10 11:50:20 +08:00
remove print
This commit is contained in:
@@ -207,7 +207,6 @@ class Attention(nn.Module):
|
||||
if not self.training and self.forward_attn_mask:
|
||||
_, n = prev_alpha.max(1)
|
||||
val, n2 = alpha.max(1)
|
||||
print(True)
|
||||
for b in range(alignment.shape[0]):
|
||||
alpha[b, n[b] + 2:] = 0
|
||||
alpha[b, :(n[b] - 1)] = 0 # ignore all previous states to prevent repetition.
|
||||
|
||||
Reference in New Issue
Block a user