mirror of https://github.com/coqui-ai/TTS.git
comments update
This commit is contained in:
parent
a012537908
commit
1c1984d25b
|
@ -9,9 +9,8 @@ from mozilla_voice_tts.tts.layers.glow_tts.glow import LayerNorm
|
||||||
|
|
||||||
|
|
||||||
class RelativePositionMultiHeadAttention(nn.Module):
|
class RelativePositionMultiHeadAttention(nn.Module):
|
||||||
"""Implementation of https://arxiv.org/pdf/1803.02155.pdf
|
"""Implementation of Relative Position Encoding based on
|
||||||
Visualization of the algorithm
|
https://arxiv.org/pdf/1809.04281.pdf
|
||||||
https://raw.githubusercontent.com/Separius/CudaRelativeAttention/master/algorithm.png
|
|
||||||
"""
|
"""
|
||||||
def __init__(self,
|
def __init__(self,
|
||||||
channels,
|
channels,
|
||||||
|
|
Loading…
Reference in New Issue