debugged

LiyuanLucasLiu · Sep 14, 2017 · c8c7831 · c8c7831
1 parent 4533996
commit c8c7831
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/model/crf.py b/model/crf.py
@@ -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)
 
@@ -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)