def __init__(self, rel_layers_setting, hidden_dim, feat_dim, pool_dim):
    super(SketchRetrievalPoolingModel, self).__init__()
    self.pool_dim = int(pool_dim)
    rel_layers_setting = rel_layers_setting.copy()
    (rel_layers_setting.insert(0, hidden_dim), rel_layers_setting.append(feat_dim))
    self.rel_fcs = []
    for i in range((len(rel_layers_setting) - 1)):
        self.rel_fcs.append(nn.Linear(rel_layers_setting[i], rel_layers_setting[(i + 1)]))
    self.rel_fcs = nn.ModuleList(self.rel_fcs)
def __init__(self, rel_layers_setting, hidden_dim, feat_dim, pool_dim):
    super(SketchRetrievalPoolingModel, self).__init__()
    self.pool_dim = int(pool_dim)
    rel_layers_setting = rel_layers_setting.copy()
    (rel_layers_setting.insert(0, hidden_dim), rel_layers_setting.append(feat_dim))
    self.rel_fcs = []
    for i in range((len(rel_layers_setting) - 1)):
        self.rel_fcs.append(nn.Linear(rel_layers_setting[i], rel_layers_setting[(i + 1)]))
    self.rel_fcs = nn.ModuleList(self.rel_fcs)