From 0afdc987a9337fc513306266b4b76cde8d09fb47 Mon Sep 17 00:00:00 2001
From: zhangxuefei <zhangxuefei@baidu.com>
Date: Mon, 26 Aug 2019 10:05:49 +0800
Subject: [PATCH] update demo text-cls.py

---
 demo/text-classification/text_classifier.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/demo/text-classification/text_classifier.py b/demo/text-classification/text_classifier.py
index 3e1e87fa..cb0b3d36 100644
--- a/demo/text-classification/text_classifier.py
+++ b/demo/text-classification/text_classifier.py
@@ -120,13 +120,8 @@ if __name__ == '__main__':
     ]
 
     if args.use_taskid:
-        feed_list = [
-            inputs["input_ids"].name,
-            inputs["position_ids"].name,
-            inputs["segment_ids"].name,
-            inputs["input_mask"].name,
-            inputs["task_ids"].name,
-        ]
+        feed_list = feed_list.append(inputs["task_ids"].name)
+
     # Select finetune strategy, setup config and finetune
     strategy = hub.AdamWeightDecayStrategy(
         weight_decay=args.weight_decay,
-- 
GitLab