ÎncepețiÎncepe gratuit

Traducere automată cu un model PyTorch pre-antrenat

Echipa ta de la PyBooks lucrează la un proiect de IA care implică traducerea dintr-o limbă în alta. Aceștia doresc să folosească modele pre-antrenate pentru această sarcină, ceea ce poate economisi mult timp și resurse de antrenament. Scopul acestui exercițiu este să configurezi un model de traducere din biblioteca Transformers de la HuggingFace — mai exact modelul T5 (Text-To-Text Transfer Transformer) — și să îl folosești pentru a traduce o frază din engleză în franceză.

T5Tokenizer și T5ForConditionalGeneration au fost deja importate pentru tine.

Acest exercițiu face parte din cursul

Deep Learning pentru text cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

  • Inițializează tokenizer și model folosind modelul pre-antrenat "t5-small".
  • Codifică textul de intrare cu ajutorul tokenizer-ului, asigurându-te că returnezi tensori PyTorch.
  • Traduce textul de intrare folosind model și generează rezultatul tradus.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Initalize tokenizer and model
tokenizer = ____.from_pretrained("t5-small")
model = ____.from_pretrained("t5-small")

input_prompt = "translate English to French: 'Hello, how are you?'"

# Encode the input prompt using the tokenizer
input_ids = ____.____(input_prompt, return_tensors="____")

# Generate the translated ouput
output = model.____(input_ids, max_length=50)
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print("Generated text:",generated_text)
Editează și rulează codul