class NetForImagePair(nn.Module):
def __init__(self):
- super(NetForImagePair, self).__init__()
+ super().__init__()
self.features_a = nn.Sequential(
nn.Conv2d(1, 16, kernel_size = 5),
nn.MaxPool2d(3), nn.ReLU(),
class NetForImageValuesPair(nn.Module):
def __init__(self):
- super(NetForImageValuesPair, self).__init__()
+ super().__init__()
self.features_a = nn.Sequential(
nn.Conv2d(1, 16, kernel_size = 5),
nn.MaxPool2d(3), nn.ReLU(),
)
def __init__(self):
- super(NetForSequencePair, self).__init__()
+ super().__init__()
self.nc = 32
self.nh = 256