Views
No views yet
transformers 庫完成訓練,並以 BLEU 分數作為主要評估指標。Seq2SeqTrainer 完成微調predict_with_generate=True(生成翻譯以供評分)training_log.csv./results/transformer_v11from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3model = AutoModelForSeq2SeqLM.from_pretrained("./results/transformer_v1")
4tokenizer = AutoTokenizer.from_pretrained("./results/transformer_v1")
5
6inputs = tokenizer("外出要小心注意安全", return_tensors="pt")
7outputs = model.generate(**inputs)
8translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
9print(translated_text)