Programming/Pytorch3 [Pytorch] 학습된 모델에서 필요한 layer 만 사용하는 방법 import torch.nn as nn import torchvision.models as models # Pre-trained ResNet 모델 불러오기 model = models.resnet50(pretrained=True) # ResNet50에서 마지막 classification layer를 제거합니다. modules = list(model.children())[:-1] model = nn.Sequential(*modules) # fully connected layer 1층 추가 model.fc = nn.Linear(2048, 1024) # 학습된 모델을 evaluation mode로 설정합니다. model.eval() 위 코드에서는 먼저 PyTorch에서 제공하는 torchvision.models.. 2023. 3. 31. [pytorch] torch.no_grad() 와 model.eval() 의 차이 참고 https://yuevelyne.tistory.com/10 [PyTorch] torch.no_grad()와 model.eval()의 차이 👋 Intro 안녕하세요, 유블린입니다. 이 글에서는 torch.no_grad()와 model.eval()의 차이점을 다뤄보려고 합니다. 머신러닝 프레임 워크인 Pytorch로 모델 학습을 진행한 뒤 모델의 Evaluation을 진행 할 때 yuevelyne.tistory.com Pytorch 로 모델 학습을 진행한 뒤 모델의 Evaluation 을 진행 할 때 model.eval() with torch.no_grad() : for batch in data_loader : ... 위와 같이 model.eval() 을 사용을 하고 torch.no_grad() 를 뒤.. 2022. 11. 30. pytorch 설치 및, GPU 설정 참조 https://chancoding.tistory.com/90 윈도우에 PyTorch 설치, GPU 설정, 자세하게 윈도우에 파이토치를 설치하고 GPU를 사용하는 방법에 대해서 알아보겠습니다. 1. 가상환경 만들기 (생략 가능) 기본적으로 저는 가상 환경을 나누어서 작업을 진행하는 것을 선호합니다. 그래서 chancoding.tistory.com 참조 https://kodorricoding.tistory.com/14 [Python] 파이썬 가상환경이란? 들어가기전에.. 파이썬을 설치하면 기본적으로 pip 패키지 매니저가 존재하고, 이를 이용해 프로젝트에 필요한 패키지를 설치할 수 있다. 이때 pip로 설치한 패키지는 전역으로 설치가 되어 모든 kodorricoding.tistory.com 1. 가상환.. 2022. 11. 24. 이전 1 다음
