신경망모델 구성 (1) 썸네일형 리스트형 [Pytorch] 신경망 모델 구성하기 파이토치 한국사용자 모임의 글을 보면서 공부한 내용을 정리한 글입니다. 신경망은 연산을 수행하는 계층/모듈로 구성된다. torch.nn 네임스페이스는 신경망을 구성하는데 필요한 모든 구성 요소를 제공한다. PyTorch의 모든 모듈은 nn.Module의 하위 클래스이다. 신경망은 다른 모듈로 구성된 모듈이다. Import import os import torch from torch import nn from torch.utils.data import DataLoader from torchvision import datasets, transforms device 세팅 device = ( "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mp.. 이전 1 다음
