update demo text-cls.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,
