from dataclasses import dataclass


@dataclass
class GSTHyperparameters:
    E = 512

    # reference encoder
    ref_enc_filters = [32, 32, 64, 64, 128, 128]

    # style token layer
    token_num = 10
    # token_emb_size = 256
    num_heads = 8

    n_mels = 256  # Number of Mel banks to generate


hparams = GSTHyperparameters()