config_model.py

# Attentional Seq2seq model.
# Hyperparameters not specified here will take the default values.

num_units = 256
beam_width = 10

embedder = {
    'dim': num_units
}

encoder = {
    'rnn_cell_fw': {
        'kwargs': {
            'num_units': num_units
        }
    }
}

decoder = {
    'rnn_cell': {
        'kwargs': {
            'num_units': num_units
        },
    },
    'attention': {
        'kwargs': {
            'num_units': num_units,
        },
        'attention_layer_size': num_units
    },
    'max_decoding_length_infer': 60,
}

opt = {
    'optimizer': {
        'type':  'Adam',
        'kwargs': {
            'lr': 0.001,
        },
    },
}