LoslegenKostenlos loslegen

Sprachübersetzung mit einem vortrainierten PyTorch-Modell

Dein Team bei PyBooks arbeitet an einem KI-Projekt, das die Übersetzung von einer Sprache in eine andere umfasst. Für diese Aufgabe sollen vortrainierte Modelle genutzt werden, was viel Trainingszeit und Ressourcen spart. In dieser Übung richtest du ein Übersetzungsmodell aus der Transformers-Bibliothek von HuggingFace ein, konkret das T5 (Text-To-Text Transfer Transformer)-Modell, und verwendest es, um einen englischen Satz ins Französische zu übersetzen.

T5Tokenizer, T5ForConditionalGeneration wurden für dich geladen.

Diese Übung ist Teil des Kurses

Deep Learning für Text mit PyTorch

Kurs anzeigen

Anleitung zur Übung

  • Initialisiere tokenizer und model aus dem vortrainierten Modell "t5-small".
  • Kodiere die Eingabeaufforderung mit dem Tokenizer und gib dabei sicher PyTorch-Tensoren zurück.
  • Übersetze die Eingabeaufforderung mit model und generiere die übersetzte Ausgabe.

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# 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)
Code bearbeiten und ausführen