Update model documentation (#933)

docs/source/models.rst

@@ -9,6 +9,14 @@ torchaudio.models
 The models subpackage contains definitions of models for addressing common audio tasks.
 
 
+:hidden:`ConvTasNet`
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: ConvTasNet
+
+  .. automethod:: forward
+
+
 :hidden:`Wav2Letter`
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 

torchaudio/models/conv_tasnet.py

@@ -22,9 +22,11 @@ class ConvBlock(torch.nn.Module):
     Note:
         This implementation corresponds to the "non-causal" setting in the paper.
 
-    References:
+    Reference:
         - Conv-TasNet: Surpassing Ideal Time--Frequency Magnitude Masking for Speech Separation
+
           Luo, Yi and Mesgarani, Nima
+
           https://arxiv.org/abs/1809.07454
     """
 
@@ -189,9 +191,11 @@ class ConvTasNet(torch.nn.Module):
     Note:
         This implementation corresponds to the "non-causal" setting in the paper.
 
-    References:
+    Reference:
         - Conv-TasNet: Surpassing Ideal Time--Frequency Magnitude Masking for Speech Separation
+
           Luo, Yi and Mesgarani, Nima
+
           https://arxiv.org/abs/1809.07454
     """