코딩 메모장

  • 홈
  • 태그
  • 방명록

CKPT 1

[Pytorch Lightning] 튜토리얼 2 - 모델을 저장하고 불러오자

이번에는 훈련한 모델을 저장하는 방법과 불러오는 방법에 대해 알아보려고 한다. 1편에서 사용한 코드를 거의 그대로 사용하며, Trainer선언 부분만 바뀌는 정도이며, 전체 코드는 깃허브에 있으니 참고하기 바란다.요약모델을 특정경로에 체크포인트 저장하기모델의 하이퍼파라미터 저장하기체크포인트로 모델을 불러오기 모델을 특정경로에 체크포인트 저장하기모델을 저장하는 것은 각 epoch마다 자동으로 저장된다. Jupyter 기준으로 설명하면, Jupyter 파일의 같은 디렉터리 내에 lightning_logs폴더에 각 validation_step마다 저장되는 것을 알 수 있다. 하지만, 나는 여기서 특정 경로에 저장하는 방법에 대해 소개하려고 한다.# 이전 코드trainer = pl.Trainer(max_epoc..

AI & DL/Pytorch Lightning 2024.07.21
이전
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

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

최근글과 인기글

  • 최근글
  • 인기글

최근댓글

공지사항

페이스북 트위터 플러그인

  • Facebook
  • Twitter

Archives

Calendar

«   2025/05   »
일 월 화 수 목 금 토
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 31

방문자수Total

  • Today :
  • Yesterday :

Copyright © Kakao Corp. All rights reserved.

티스토리툴바