def __init__(self, out_dim, layers_dim):
    super(SketchEmbeddingRefineNetwork, self).__init__()
    self.layers = []
    layers_dim = layers_dim.copy()
    layers_dim.append(out_dim)
    for i in range((len(layers_dim) - 1)):
        self.layers.append(nn.Linear(layers_dim[i], layers_dim[(i + 1)]))
    self.layers = nn.ModuleList(self.layers)
def __init__(self, out_dim, layers_dim):
    super(SketchEmbeddingRefineNetwork, self).__init__()
    self.layers = []
    layers_dim = layers_dim.copy()
    layers_dim.append(out_dim)
    for i in range((len(layers_dim) - 1)):
        self.layers.append(nn.Linear(layers_dim[i], layers_dim[(i + 1)]))
    self.layers = nn.ModuleList(self.layers)