코딩 메모장

  • 홈
  • 태그
  • 방명록

module 1

[Pytorch Lightning] BART를 훈련해 Text를 요약해보자 - 1

1편 : 프로젝트 요약 및 LightningModule 설계2편 : Dataset 구축 및 DataLoader 구축3편 : logger 작성, wandb 연동 및 확인4편 : 모델 로딩 및 실행결과(wandb) 확인  오늘은 이전에 캡스톤디자인에서 BART를 이용해서 Text를 요약하는 Task를 진행했었는데, 이것에 대해 설명하면서 코드를 어떻게 작성했는지 설명하려고 한다. 간단하게 요약하면 다음과 같다. 카페 메뉴에 대한 Text가 입력이 되면, BART를 이용해서 영어로 번역한 뒤 필요한 내용만 출력하는 것이다. 아이스 아메리카노 하나, 카페라떼 하나를 영어로 요약하는 것이다. 다음과 같은 Task를 위해 BART를 훈련시키는 코드에 대해 설명하겠다. 필요한 라이브러리 설치pip install da..

AI & DL/Pytorch Lightning 2024.07.11
이전
1
다음
더보기
250x250
프로필사진

코딩 메모장

  • 분류 전체보기 (149)
    • Paper (25)
      • Metric (6)
      • Model (3)
    • AI & DL (23)
      • Pytorch (6)
      • Hugging Face (7)
      • Pytorch Lightning (8)
      • Lang Chain (0)
    • Coding (35)
      • Python (25)
      • Numpy (5)
      • Clean Code (3)
      • SQL (1)
    • Algorithm (45)
      • BOJ (25)
      • 프로그래머스 (17)
    • Django (1)
    • Cloud (1)
      • GCP (1)
    • Program (7)
      • Docker (5)
      • VSCode (2)
    • AI API (3)
      • OpenAI (2)
      • Gemini (1)
    • Project (1)
    • 활동 (7)
      • 교내활동 (4)
      • 대외활동 (2)
      • 잡동사니 (1)

Tag

DataSet, PyTorch Lightning, Silver, pytorch, nlp, BOJ, numpy, docker, 프로그래머스, Python, Metric, 실버, LEVEL 2, ESconv, 백준, 오블완, c++, paper, huggingface, 파이썬,

최근글과 인기글

  • 최근글
  • 인기글

최근댓글

공지사항

페이스북 트위터 플러그인

  • Facebook
  • Twitter

Archives

Calendar

«   2025/06   »
일 월 화 수 목 금 토
1 2 3 4 5 6 7
8 9 10 11 12 13 14
15 16 17 18 19 20 21
22 23 24 25 26 27 28
29 30

방문자수Total

  • Today :
  • Yesterday :

Copyright © Kakao Corp. All rights reserved.

티스토리툴바