def __init__(self, args):
super(res_SN, self).__init__()
self.batch_size = args[0]
self.cuda = args[1]
b = self.batch_size
cuda = self.cuda
self.res = resnet50()
self.sn = SN(b, 24, 8, 8, 2, cuda=cuda, method=1)
def __init__(self, args):
super(res_RN_FC, self).__init__()
self.batch_size = args[0]
self.cuda = args[1]
b = self.batch_size
cuda = self.cuda
self.res = resnet50()
self.rn = RN(b, 24, 8, 8, 64, cuda=cuda, method=0)
self.fc = FC(24, 8, 8, 64, cuda)
self.final = nn.Sequential(nn.Linear(128, 64), nn.ReLU(),
nn.Linear(64, 32), nn.ReLU(),
nn.Linear(32, 2))
def __init__(self, args):
super(res_TRSF_TRN, self).__init__()
self.resnet = resnet50()
self.trsf_trn = TRSF_TRN(args, (24, 8, 8), 2)
def __init__(self, args):
super(res_TRSF, self).__init__()
self.resnet = resnet50()
self.trsf = TRSF(args, (24, 8, 8), 2)
