提交 84b997b9 编写于 作者: S Steffy-zxf 提交者: wuzewu

Update the seq-label-task fetch_list (#52)

上级 bc2c5e71
......@@ -859,6 +859,14 @@ class SequenceLabelTask(BasicTask):
feed_list += [self.seq_len.name]
return feed_list
@property
def fetch_list(self):
if self.is_train_phase or self.is_test_phase:
return [metric.name for metric in self.metrics] + [self.loss.name]
elif self.is_predict_phase:
return [self.ret_infers.name] + [self.seq_len.name]
return [self.output.name]
class MultiLabelClassifierTask(ClassifierTask):
def __init__(self,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册