def __init__(self, hidden_dim, M, cls_in_input, rel_in_input):
    super(MaskSketchGMMModel, self).__init__()
    self.re_fc = nn.Linear(hidden_dim, ((6 * M) + 3))
    self.cls_in_input = cls_in_input
    self.rel_in_input = rel_in_input
def __init__(self, hidden_dim, M, cls_in_input, rel_in_input):
    super(MaskSketchGMMModel, self).__init__()
    self.re_fc = nn.Linear(hidden_dim, ((6 * M) + 3))
    self.cls_in_input = cls_in_input
    self.rel_in_input = rel_in_input