config_sp = {
    "D_MODEL":  256,
    "D_FF": 1024,
    "NUM": 3,
    "NUM_HEADS": 4,
    "MAX_LEN": 32,
    }

config_g2p = {
        "D_MODEL":  512,
        "D_FF": 2048,
        "NUM": 6,
        "NUM_HEADS": 8,
        "MAX_LEN": 32,
        }