def __init__(self, cls_layers_setting, hidden_dim, cls_dim, pool_dim):
    super(SketchClsPoolingModel, self).__init__()
    self.pool_dim = int(pool_dim)
    cls_layers_setting = cls_layers_setting.copy()
    (cls_layers_setting.insert(0, hidden_dim), cls_layers_setting.append(cls_dim))
    self.cls_fcs = []
    for i in range((len(cls_layers_setting) - 1)):
        self.cls_fcs.append(nn.Linear(cls_layers_setting[i], cls_layers_setting[(i + 1)]))
    self.cls_fcs = nn.ModuleList(self.cls_fcs)
def __init__(self, cls_layers_setting, hidden_dim, cls_dim, pool_dim):
    super(SketchClsPoolingModel, self).__init__()
    self.pool_dim = int(pool_dim)
    cls_layers_setting = cls_layers_setting.copy()
    (cls_layers_setting.insert(0, hidden_dim), cls_layers_setting.append(cls_dim))
    self.cls_fcs = []
    for i in range((len(cls_layers_setting) - 1)):
        self.cls_fcs.append(nn.Linear(cls_layers_setting[i], cls_layers_setting[(i + 1)]))
    self.cls_fcs = nn.ModuleList(self.cls_fcs)