赞
踩
比如用T5模型,训练数据是inputText-outputText格式,预测时do_sample=False
# 预测代码
from transformers import TFAutoModelForSeq2SeqLM
from transformers import AutoTokenizer
checkpoint_local = "./path/"
tokenizer = AutoTokenizer.from_pretrained(checkpoint_local)
model = TFAutoModelForSeq2SeqLM.from_pretrained(checkpoint_local,
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。