def __init__(self, node_id, num_prev_nodes, channels, num_downsample_nodes):
    super().__init__()
    edges = nn.ModuleDict()
    for i in range(num_prev_nodes):
        if (i < num_downsample_nodes):
            stride = 2
        else:
            stride = 1
        edge_id = '{}_p{}'.format(node_id, i)
        edges.add_module(edge_id, nn.Sequential(Placeholder(group='node', space_id=edge_id, choice_args=dict(stride=stride, in_channels=channels, out_channels=channels))))
    self.edges = Placeholder(group='node_edge', space_id=node_id, choices=edges)
def __init__(self, node_id, num_prev_nodes, channels, num_downsample_nodes):
    super().__init__()
    edges = nn.ModuleDict()
    for i in range(num_prev_nodes):
        if (i < num_downsample_nodes):
            stride = 2
        else:
            stride = 1
        edge_id = '{}_p{}'.format(node_id, i)
        edges.add_module(edge_id, nn.Sequential(Placeholder(group='node', space_id=edge_id, choice_args=dict(stride=stride, in_channels=channels, out_channels=channels))))
    self.edges = Placeholder(group='node_edge', space_id=node_id, choices=edges)