X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=causal-autoregression.py;h=77542655fa5fb958bcdb6d13927bd1add205b21a;hp=c2f6161511b08c56253f224535c31f15b8946fcc;hb=d74d7be5abef26c78d014bd179f2c52f81aca65b;hpb=762a2c5e2485e0ebd7c26fe980893a4de2544bb9

diff --git a/causal-autoregression.py b/causal-autoregression.py
index c2f6161..7754265 100755
--- 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)