Skip to content

Commit

Permalink
debugged
Browse files Browse the repository at this point in the history
  • Loading branch information
LiyuanLucasLiu committed Sep 14, 2017
1 parent 4533996 commit c8c7831
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions model/crf.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def forward(self, feats):
args:
feats (batch_size, seq_len, hidden_dim) : input score from previous layers
return:
Tensor (batch_size, seq_len, tag_size, tag_size) : output from crf layer
output from crf layer (batch_size, seq_len, tag_size, tag_size)
"""
return self.hidden2tag(feats).view(-1, self.tagset_size, self.tagset_size)

Expand Down Expand Up @@ -70,8 +70,7 @@ def forward(self, feats):
args:
feats (batch_size, seq_len, hidden_dim) : input score from previous layers
return:
Tensor (batch_size, seq_len, tag_size, tag_size) : output from crf layer
output from crf layer (batch_size, seq_len, tag_size, tag_size)
"""

ins_num = feats.size(0)
Expand Down

0 comments on commit c8c7831

Please sign in to comment.