Edit model card

en-zhtw

English-to-Traditional Chinese sentence translator

This model is a fine-tuned version of Helsinki-NLP/opus-mt-en-zh on the zetavg/coct-en-zh-tw-translations-twp-300k dataset.

This is so it can output Traditional Chinese by default and make the translations more natural sounding.

Model description

  • input: English text only
  • output: Traditional Chinese text translation

How to use:

from transformers import pipeline

model_checkpoint = "agentlans/en-zhtw"
translator = pipeline("translation", model=model_checkpoint)

translator(
    [
        "Even if you spend a day in Windsor you'll notice that it's a very multicultural city, yet still retaining a small town feel.",
        "Its main waterfront park stretches about 5 km (3.1 mi), from the 1929 Ambassador suspension bridge past the contemporary Windsor Sculpture Park.",
    ]
)

# [{'translation_text': '儘管在風沙住了一天,都會發現這裡是個非常多樣化的城市,但還是保留了一個小鎮的感覺。'}, {'translation_text': '從 1929 年的大使吊橋到今天的風雕公園,總長約 5 公里。'}]

Intended uses & limitations

  • English to Traditional Chinese translation
  • Single sentence
  • Limitations: may hallucinate or omit information, doesn't understand context, can still sound awkward or strange (as the above example shows)

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 3.0

Framework versions

  • Transformers 4.43.3
  • Pytorch 2.3.0+cu121
  • Datasets 2.20.0
  • Tokenizers 0.19.1
Downloads last month
12
Safetensors
Model size
77.5M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for agentlans/en-zhtw

Finetuned
(20)
this model

Dataset used to train agentlans/en-zhtw