config.json

{
    "N_encoder" : 4,
    "N_heads_enc" : 1,

    "N_decoder" : 4,
    "N_heads_dec" : 4,

    "Gen_Embed" : 128,

    "Forward_Expansion" : 64,
    "Dropout" : 0.0,

    "DiagAttn" : 400,
    "PhonemeNoiseProb" : 0.0,

    "epochs" : 200,
    "learn_rate" : 1e-4,
    "batch_size" : 8,

    "Training" : true,
    "Testing" : true,

    "Device" : "cpu",

    "MainPath" : "/data1/Code/Siddarth/Code/Upload/"
}