def __init__(self, input_channel, pre_input_channel, out_channel, reduction_prev=False):
super(AmoebaNet_RC, self).__init__()
#print(C_prev_prev, C_prev, C)
genotype = eval("genotypes.%s" % "AmoebaNet")
concat = genotype.reduce_concat
C_prev_prev, C_prev, C = pre_input_channel, input_channel, (out_channel//len(concat))//2*2
reduction, reduction_prev = True, reduction_prev
self.Cell = AddConcetCell(genotype, C_prev_prev, C_prev, C, reduction, reduction_prev)
self.out_dim = C*len(concat)
self.depth = 5
self.density_edge = 23
self.density_node = 16
def __init__(self, input_channel, pre_input_channel, out_channel, reduction_prev=False):
super(RENAS_NC, self).__init__()
#print(C_prev_prev, C_prev, C)
genotype = eval("genotypes.%s" % "RENAS")
concat = genotype.normal_concat
C_prev_prev, C_prev, C = pre_input_channel, input_channel, (out_channel//len(concat))//2*2
reduction, reduction_prev = False, reduction_prev
self.Cell = AddConcetCell(genotype, C_prev_prev, C_prev, C, reduction, reduction_prev)
self.out_dim = C*len(concat)
self.depth = 5
self.density_edge = 24
self.density_node = 16
def __init__(self,
input_channel,
pre_input_channel,
out_channel,
reduction_prev=False):
super(GDAS_V2_RC, self).__init__()
genotype = eval("genotypes.%s" % "GDAS_V2")
concat = genotype.reduce_concat
C_prev_prev, C_prev, C = pre_input_channel, input_channel, (
out_channel // len(concat)) // 8 * 8
reduction, reduction_prev = True, reduction_prev
self.Cell = AddConcetCell(genotype, C_prev_prev, C_prev, C, reduction,
reduction_prev)
self.out_dim = C * len(concat)
self.depth = 2
self.density_edge = 8
self.density_node = 5