AI 반도체 노트

  • 홈
  • 태그
  • 방명록

pytorch 1

10. Save와 Load

0. 한 문장으로 이해하기PyTorch에서는 모델 전체 객체보다 state_dict를 저장하는 방식을 기본으로 익히는 것이 좋습니다.state_dict는 모델의 학습 파라미터와 일부 layer가 등록한 buffer 값을 이름과 Tensor로 담은 딕셔너리입니다. 예를 들어 BatchNorm의 running_mean 같은 값도 포함될 수 있습니다.1. 모델 저장PATH = 'model.pth'torch.save(model.state_dict(), PATH)이 코드는 모델의 학습된 파라미터를 저장합니다.2. 모델 불러오기저장한 파라미터를 불러오려면 먼저 같은 모델 구조를 만들어야 합니다.model = NeuralNetwork()state_dict = torch.load(PATH, weights_only=Tr..

인공지능 코딩/PyTorch 2026.06.23
이전
1
다음
더보기
프로필사진

AI 반도체 노트

인공지능으로 반도체를 설계해봐요. 블로그에서는 기본적인 이론과 내용부터 다룹니다.

  • 카테고리 (30)
    • 반도체 (1)
      • 디지털회로설계 (1)
      • Physical Design (0)
    • 인공지능 이론 (11)
      • Diffusion (7)
      • Graph with Machine Learning (4)
    • 인공지능 코딩 (14)
      • PyTorch (14)
    • 비전공자도 이해하는 기초 CS (2)
      • 비전공자도 이해하는 자료구조 (2)
      • 비전공자도 이해하는 알고리즘 (0)
      • 기초 CS 상식 (0)

Tag

인공지능, 딥러닝, generativeAI, 인공지능학습, 생성형 모델, graphs, Diffusion, pytorch, Undirected, Ai, 디퓨전, Graph Representation, adjency list, 해시테이블, 나노바나나, llm, 딥러닝실습, 파이토치, flowmatching, 인공지능실습,

최근글과 인기글

  • 최근글
  • 인기글

최근댓글

공지사항

페이스북 트위터 플러그인

  • Facebook
  • Twitter

Archives

Calendar

«   2026/07   »
일 월 화 수 목 금 토
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 © AXZ Corp. All rights reserved.

티스토리툴바