diff --git a/wavenet_model.py b/wavenet_model.py index 3ecdd2761..95301f809 100644 --- a/wavenet_model.py +++ b/wavenet_model.py @@ -130,13 +130,13 @@ def wavenet(self, input, dilation_func): # WaveNet layers for i in range(self.blocks * self.layers): - # |----------------------------------------| *residual* + # |----------------------------------------| *residual* # | | - # | |-- conv -- tanh --| | + # | |-- conv -- tanh --| | # -> dilate -|----| * ----|-- 1x1 -- + --> *input* - # |-- conv -- sigm --| | - # 1x1 - # | + # |-- conv -- sigm --| | + # 1x1 + # | # ---------------------------------------> + -------------> *skip* (dilation, init_dilation) = self.dilations[i]