提交 d4938b7d 编写于 作者: W wuzewu

Update task api

上级 dc9f9650
...@@ -417,9 +417,6 @@ class BasicTask(object): ...@@ -417,9 +417,6 @@ class BasicTask(object):
def _build_env_end_event(self): def _build_env_end_event(self):
pass pass
def _calculate_metrics(self, run_states):
raise NotImplementedError
def _eval_start_event(self): def _eval_start_event(self):
logger.info("Evaluation on {} dataset start".format(self.phase)) logger.info("Evaluation on {} dataset start".format(self.phase))
...@@ -461,6 +458,9 @@ class BasicTask(object): ...@@ -461,6 +458,9 @@ class BasicTask(object):
def _add_metrics(self): def _add_metrics(self):
raise NotImplementedError raise NotImplementedError
def _calculate_metrics(self, run_states):
raise NotImplementedError
# NOTE: current saved checkpoint machanism is not completed, # NOTE: current saved checkpoint machanism is not completed,
# it can't restore dataset training status # it can't restore dataset training status
def save_checkpoint(self, epoch, step): def save_checkpoint(self, epoch, step):
...@@ -946,10 +946,10 @@ class SequenceLabelTask(BasicTask): ...@@ -946,10 +946,10 @@ class SequenceLabelTask(BasicTask):
class MultiLabelClassifierTask(ClassifierTask): class MultiLabelClassifierTask(ClassifierTask):
def __init__(self, def __init__(self,
data_reader,
feature, feature,
num_classes, num_classes,
feed_list, feed_list,
data_reader,
startup_program=None, startup_program=None,
config=None, config=None,
hidden_units=None): hidden_units=None):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册