def __init__(self, word_num, feature_num, hidden_num):
super(ImageCaption, self).__init__(
word_vec = L.EmbedID(word_num, hidden_num),
image_vec = L.Linear(feature_num, hidden_num),
lstm = L.LSTM(hidden_num, hidden_num),
out_word = L.Linear(hidden_num, word_num),
)
评论列表
文章目录