1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 텍스트 딥러닝
Connected

연습 문제

사전 학습된 PyTorch 모델로 언어 번역하기

PyBooks 팀은 한 언어에서 다른 언어로 번역하는 AI 프로젝트를 진행하고 있어요. 이 작업에 사전 학습된 모델을 활용하면 학습 시간과 자원을 크게 절약할 수 있습니다. 이번 연습 문제의 목표는 HuggingFace Transformers 라이브러리의 T5(Text-To-Text Transfer Transformer) 모델을 설정하고, 이를 사용해 영어 문장을 프랑스어로 번역하는 것입니다.

T5Tokenizer, T5ForConditionalGeneration는 미리 불러와 두었습니다.

지침

100 XP
  • 사전 학습된 "t5-small" 모델에서 tokenizer와 model을 초기화하세요.
  • 토크나이저를 사용해 입력 프롬프트를 인코딩하고, PyTorch 텐서를 반환하도록 설정하세요.
  • model을 사용해 입력 프롬프트를 번역하고, 번역된 출력을 생성하세요.