search

TobiasLee / MTA-LSTM-TensorFlow

TensorFlow reimplementation of Topic-to-Essay Generation with Neural Networks.
https://tobiaslee.top/2018/11/02/customized-RNN-cell/
70 stars 24 forks source link

Config.py #10

Closed ardand closed 4 years ago

ardand commented 4 years ago

When using MTA model how to set the parameter 'attention_size' ?

TobiasLee commented 4 years ago

modify the attention param in the topic_wrapper.py

class TopicAttentionWrapper(RNNCell):
    def __init__(self, cell, memory, attention_size=128)