torch.nn1 [파이토치] 신경망 모델 구성/Autograd/최적화 파이토치 한국어 튜토리얼에서 '신경망 모델 구성하기', 'Autograd', '최적화' 부분에 대해서 공부하고 정리함 1. 신경망 모델 구성하기 import os import torch from torch import nn from torch.utils.data import DataLoader from torchvision import datasets, transforms 학습을 얻기 위한 장치 device = ( "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu" ) print(f"Using {device} device") -----------------------------------.. 2023. 5. 14. 이전 1 다음