def __init__(self, layers_setting, input_dim, cls_dim, max_length, position_type, attention_norm_type, inter_activation, attention_dropout_prob, hidden_dropout_prob, output_dropout_prob):
    super(SketchGANGenerator, self).__init__()
    self.encoder = SketchTransformerModel(layers_setting, input_dim, cls_dim, max_length, position_type, attention_norm_type, inter_activation, attention_dropout_prob, hidden_dropout_prob, output_dropout_prob)
    self.output = nn.Linear(layers_setting[0][1], 5)
def __init__(self, layers_setting, input_dim, cls_dim, max_length, position_type, attention_norm_type, inter_activation, attention_dropout_prob, hidden_dropout_prob, output_dropout_prob):
    super(SketchGANGenerator, self).__init__()
    self.encoder = SketchTransformerModel(layers_setting, input_dim, cls_dim, max_length, position_type, attention_norm_type, inter_activation, attention_dropout_prob, hidden_dropout_prob, output_dropout_prob)
    self.output = nn.Linear(layers_setting[0][1], 5)