def __init__(self, config):
super(Encoder, self).__init__()
self.config = config
input_size = config.d_proj if config.projection else config.d_embed
dropout = 0 if config.n_layers == 1 else config.dp_ratio
self.rnn = nn.LSTM(input_size=input_size,
hidden_size=config.d_hidden,
num_layers=config.n_layers,
dropout=dropout,
bidirectional=config.birnn)
def __init__(self, config):
super(Encoder, self).__init__()
#self.config = config
input_size = config["d_proj"] if config[
"projection"] else config["d_embed"]
dropout = 0 if config["n_layers"] == 1 else config["dp_ratio"]
self.rnn = nn.LSTM(input_size=input_size,
hidden_size=config["d_hidden"],
num_layers=config["n_layers"],
dropout=dropout,
bidirectional=config["birnn"])
self.n_cells = config["n_cells"]
self.d_hidden = config["d_hidden"]
self.birnn = config["birnn"]