update format

Author: Ji Chen

test/test_models.py

@@ -58,7 +58,7 @@ def test_waveform(self):
 class TestUpsampleNetwork(common_utils.TorchaudioTestCase):
 
     def test_waveform(self):
-        """test the output dimensions after _UpsampleNetwork block.
+        """Validate the output dimensions of a _UpsampleNetwork block.
         """
 
         upsample_scales = [5, 5, 8]

torchaudio/models/_wavernn.py

@@ -100,8 +100,8 @@ class _Stretch2d(nn.Module):
     Examples
         >>> stretch2d = _Stretch2d(time_scale=10, freq_scale=5)
 
-        >>> input = torch.rand(10, 100, 512)
-        >>> output = stretch2d(input) # shape: (10, 500, 5120)
+        >>> input = torch.rand(10, 100, 512)  # a random spectrogram
+        >>> output = stretch2d(input)  # shape: (10, 500, 5120)
     """
 
     def __init__(self,
@@ -137,8 +137,8 @@ class _UpsampleNetwork(nn.Module):
 
     Examples
         >>> upsamplenetwork = _UpsampleNetwork(upsample_scales=[4, 4, 16])
-        >>> input = torch.rand(10, 128, 10)
-        >>> output = upsamplenetwork(input) # shape: (10, 1536, 128), (10, 1536, 128)
+        >>> input = torch.rand(10, 128, 10)  # a random spectrogram
+        >>> output = upsamplenetwork(input)  # shape: (10, 1536, 128), (10, 1536, 128)
     """
 
     def __init__(self,