작성자: 13기 김현지 Pytorch로 신경망 모델 정의하고 사용하기 Reference: 파이토치 사용자 모임-튜토리얼 anweh 티스토리: https://anweh.tistory.com/21?category=946891, https://anweh.tistory.com/22?category=94 Contents 신경망 모델 구성하기 모델 매개변수 최적화하기 모델 저장하고 불러오기 1. 신경망 모델 구성하기 FashionMNST 데이터셋의 이미지들을 분류하는 신경망을 구성해보자. import os import torch from torch import nn from torch.utils.data import DataLoader from torchvision import datasets, transforms..

알쓸쿠잡/프레임워크 2021. 11. 24. 14:20

작성자: 13기 김현지 PyTorh 사용법 정리! Reference: PyTorch로 시작하는 딥러닝 입문, 파이토치 한국 사용자 모임-튜토리얼 Contents 파이토치 패키지의 기본 구성 DATASET과 DATALOADER 파이토치 패키지의 기본 구성 1. torch 메인 네임스페이스. 텐스 등의 다양한 수학 함수가 포함되어져 있으며 Numpy와 유사한 구조를 가진다. 2. torch.autograd 자동 미분을 위한 함수들이 포함되어 있다. 자동 미분의 on/off를 제어하는 콘텍스트 매니저(enable_grad/no_grad)나 자체 미분 가능 함수를 정의할 때 사용하는 기반 클래스인 'Function' 등이 포함되어 있다. 3. torch.nn 신경망을 구축하기 위한 다양한 데이터 구조나 레이어 ..

알쓸쿠잡/프레임워크 2021. 11. 24. 14:16