def __init__(self, hidden_dim, cls_dim, max_length):
    super(SketchClassificationModel, self).__init__()
    self.fc1 = nn.Linear(hidden_dim, 4)
    self.fc2 = nn.Linear((max_length * 4), cls_dim)
def __init__(self, hidden_dim, cls_dim, max_length):
    super(SketchClassificationModel, self).__init__()
    self.fc1 = nn.Linear(hidden_dim, 4)
    self.fc2 = nn.Linear((max_length * 4), cls_dim)