Simplified calls to superclass constructors.

[pytorch.git] / causal-autoregression.py

diff --git a/causal-autoregression.py b/causal-autoregression.py
index c2f6161..7754265 100755 (executable)
--- a/causal-autoregression.py
+++ b/causal-autoregression.py
@@ -85,7 +85,7 @@ else:
 
 class NetToy1d(nn.Module):
     def __init__(self, nb_classes, ks = 2, nc = 32):
-        super(NetToy1d, self).__init__()
+        super().__init__()
         self.pad = (ks - 1, 0)
         self.conv0 = nn.Conv1d(1, nc, kernel_size = 1)
         self.conv1 = nn.Conv1d(nc, nc, kernel_size = ks)
@@ -105,7 +105,7 @@ class NetToy1d(nn.Module):
 
 class NetToy1dWithDilation(nn.Module):
     def __init__(self, nb_classes, ks = 2, nc = 32):
-        super(NetToy1dWithDilation, self).__init__()
+        super().__init__()
         self.conv0 = nn.Conv1d(1, nc, kernel_size = 1)
         self.pad1 = ((ks-1) * 2, 0)
         self.conv1 = nn.Conv1d(nc, nc, kernel_size = ks, dilation = 2)
@@ -130,7 +130,7 @@ class NetToy1dWithDilation(nn.Module):
 
 class PixelCNN(nn.Module):
     def __init__(self, nb_classes, in_channels = 1, ks = 5):
-        super(PixelCNN, self).__init__()
+        super().__init__()
 
         self.hpad = (ks//2, ks//2, ks//2, 0)
         self.vpad = (ks//2,     0,     0, 0)