Downloads · 30 days
5
16% of all-time downloads
t5-3b-wmt16-ro-en
snowfly/t5-3b-wmt16-ro-en
t5-3b-wmt16-ro-en is a machine learning model from snowfly. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
- 使用模型：t5-3b - 使用数据：wmt16 ro-en（共数据610320，使用了其中2000） - 使用分布式工具：deepspeed - 显卡：一张RTX 4090,24G - 目标：训练模型在ro-en上的翻译能力
All-time downloads
31
Public
Parameters
3B
11.8 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors11.8 GB · 100%
At a glance
- Library
- transformers
- License
- apache-2.0
- Model type
- t5
- Access
- Public
- Created
- Jun 16, 2024
- Updated
- Jun 16, 2024
- SHA
- e546f3c1
Datasets
Model card
From the Hugging Face model README
模型介绍
- 使用模型：t5-3b
- 使用数据：wmt16 ro-en（共数据610320，使用了其中2000）
- 使用分布式工具：deepspeed
- 显卡：一张RTX 4090,24G
- 目标：训练模型在ro-en上的翻译能力
使用方法
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
t5_qa_model = AutoModelForSeq2SeqLM.from_pretrained("snowfly/t5-3b-wmt16-ro-en")
t5_tok = AutoTokenizer.from_pretrained("snowfly/t5-3b-wmt16-ro-en")
input_ids = t5_tok("When was Franklin D. Roosevelt born?",
return_tensors="pt").input_ids
gen_output = t5_qa_model.generate(input_ids)[0]
print(t5_tok.decode(gen_output, skip_special_tokens=True))
- Library
- transformers
- Type
- t5
- License
- apache-2.0
- Languages
- en, ro
- Created
- Jun 16, 2024
- Updated
- Jun 16, 2024