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)
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)
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)