From 89533176138ec3c4f65fd871cbdf68695eb8e829 Mon Sep 17 00:00:00 2001 From: LiyuanLucasLiu Date: Thu, 14 Sep 2017 15:13:06 -0500 Subject: [PATCH] debugged --- model/crf.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/model/crf.py b/model/crf.py index ac1d5f6..a3d2a01 100644 --- a/model/crf.py +++ b/model/crf.py @@ -19,16 +19,9 @@ class CRF_L(nn.Module): This CRF layer is used in Ma et al. 2016, has more parameters than CRF_S arg: - hidden_dim : input dim size - tagset_size: target_set_size - if_biase: whether allow bias in linear trans - - inpug: - - **feats** (batch_size, seq_len, hidden_dim) : input score from previous layers - - output: - - **crf_score** (batch_size, seq_len, tag_size, tag_size) : output from crf layer - + hidden_dim : input dim size | + tagset_size: target_set_size + |if_biase: whether allow bias in linear trans """ @@ -43,6 +36,13 @@ def rand_init(self): utils.init_linear(self.hidden2tag) def forward(self, feats): + """ + inpug: + - **feats** (batch_size, seq_len, hidden_dim) : input score from previous layers + + output: + - **crf_score** (batch_size, seq_len, tag_size, tag_size) : output from crf layer + """ return self.hidden2tag(feats).view(-1, self.tagset_size, self.tagset_size)