From f8c5cad27c258b898a5b3637a59926280705a6e2 Mon Sep 17 00:00:00 2001 From: wuzewu Date: Mon, 10 Jun 2019 14:59:34 +0800 Subject: [PATCH] Update task API --- paddlehub/finetune/task.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddlehub/finetune/task.py b/paddlehub/finetune/task.py index e840fc6c..0fbb375c 100644 --- a/paddlehub/finetune/task.py +++ b/paddlehub/finetune/task.py @@ -554,10 +554,10 @@ class BasicTask(object): class ClassifierTask(BasicTask): def __init__(self, - data_reader, feature, num_classes, feed_list, + data_reader, startup_program=None, config=None, hidden_units=None): @@ -662,10 +662,10 @@ ImageClassifierTask = ClassifierTask class TextClassifierTask(ClassifierTask): def __init__(self, - data_reader, feature, num_classes, feed_list, + data_reader, startup_program=None, config=None, hidden_units=None): @@ -711,8 +711,8 @@ class SequenceLabelTask(BasicTask): feature, max_seq_len, num_classes, - data_reader, feed_list, + data_reader, startup_program=None, config=None, ): -- GitLab