잡다한 포스팅

  • 홈
  • 태그
  • 방명록

모델 불러오기 1

(Pytorch) Pytorch를 이용하여 학습한 모델 저장/불러오는 방법

Pytorch를 이용하여 Model을 저장하는 방법은 아래와 같습니다. import torch import torch.nn as nn device = 'cuda' if torch.cuda.is_available() else 'cpu' # CNN_model 예시 class CNN_model(nn.Module): #tistory 코드 블럭의 문제인지 indent가 맞질 않습니다... def __init__(self): ... ... def forward(self, x): ... ... model = CNN_model() # torch.save(model, path_dict_file_name) # model 전체 저장 torch.save(model, 'model.pt') # state_dict = 학습 가능한..

python 2022.04.07
1
더보기
프로필사진

인공지능 관련 공부한 자료 / 일상 / 잡다한 포스팅 블로그입니다.

공지사항

  • 분류 전체보기 (9)
    • 소개소개 (0)
    • AI (5)
    • python (2)
    • 식도락 (2)

Tag

Patch GANs, model save, weights, 모델 저장, model load, python crawling, CycleGAN, overlap tile, Multiple Linear Regression, Semantic Segmentation, load_state_dict, mirroring extrapolate, 선셋레코드, 모델 불러오기, segmentation, U-Net, Grdient Descent, state_dict, pytorch, U-Net Model,

최근글과 인기글

  • 최근글
  • 인기글

최근댓글

Archives

Calendar

«   2025/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 © Kakao Corp. All rights reserved.

티스토리툴바