def __init__(self):
super(AvgModel, self).__init__()
self.conv1 = utils.GraphConv1x1(6, 128, batch_norm=None)
for i in range(15):
module = utils.AvgResNet2(128)
self.add_module("rn{}".format(i), module)
self.conv2 = utils.GraphConv1x1(128, 120, batch_norm="pre")
def __init__(self):
super(AvgModel, self).__init__()
self.conv1 = utils.GraphConv1x1(3, 64, batch_norm=None)
for i in range(5):
module = utils.AvgResNet2(64)
self.add_module("rn{}".format(i), module)
self.bn_conv2 = utils.GraphConv1x1(64, 64, batch_norm="pre")
self.fc1 = nn.Linear(64, 10)
def __init__(self, layer):
super().__init__()
self.conv1 = utils.GraphConv1x1(3, 128, batch_norm=None)
self.layer = layer
for i in range(self.layer):
if i % 2 == 0:
module = utils.LapResNet2(128)
else:
module = utils.AvgResNet2(128)
self.add_module("rn{}".format(i), module)
self.conv2 = utils.GraphConv1x1(128, 120, batch_norm="pre")
def __init__(self):
super(DirModel, self).__init__()
self.conv1 = utils.GraphConv1x1(6, 128, batch_norm=None)
for i in range(15):
if i % 2 == 0:
module = utils.DirResNet2(128)
else:
module = utils.AvgResNet2(128)
self.add_module("rn{}".format(i), module)
self.do = nn.Dropout2d()
self.conv2 = utils.GraphConv1x1(128, 120, batch_norm="pre")