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

update demo text-cls.py

上级 df308352
...@@ -120,13 +120,8 @@ if __name__ == '__main__': ...@@ -120,13 +120,8 @@ if __name__ == '__main__':
] ]
if args.use_taskid: if args.use_taskid:
feed_list = [ feed_list = feed_list.append(inputs["task_ids"].name)
inputs["input_ids"].name,
inputs["position_ids"].name,
inputs["segment_ids"].name,
inputs["input_mask"].name,
inputs["task_ids"].name,
]
# Select finetune strategy, setup config and finetune # Select finetune strategy, setup config and finetune
strategy = hub.AdamWeightDecayStrategy( strategy = hub.AdamWeightDecayStrategy(
weight_decay=args.weight_decay, weight_decay=args.weight_decay,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册