def __init__(self, n_layers, n_units, width=3, dropout=0.2):
super(ConvGLUEncoder, self).__init__()
links = [('l{}'.format(i + 1),
ConvGLU(n_units, width=width, dropout=dropout))
for i in range(n_layers)]
for link in links:
self.add_link(*link)
self.conv_names = [name for name, _ in links]
评论列表
文章目录