提交 0afdc987 编写于 作者: Z zhangxuefei

update demo text-cls.py

上级 df308352
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册