AI/Pytorch (3) 썸네일형 리스트형 Pytorch collate_fn 이란? DataLoader에는 여러 파라미터가 있어 필요시 적절한 파라미터를 활용해 여러 설정을 줄 수 있다. 그 중에서도 collate_fn은 variable length가 달라서, 패딩해줄 때 사용한다. from torch.utils.data import Dataset, DataLoader import torch class ExampleDataset(Dataset): def __init__(self, num): self.num = num def __len__(self): return self.num def __getitem__(self, idx): return {"X":torch.tensor([idx] * (idx+1), dtype=torch.float32), "y": torch.tensor(idx, dty.. [Pytorch] 파이토치 모델 정의, 사전 학습 모델 모델 정의 nn.Module 상속 클래스 정의 nn.Module을 상속받는 클래스 정의 __init__(): 모델에서 사용될 모듈과 활성화 함수 등을 정의 forward(): 모델에서 실행되어야 하는 연산을 정의 from torch import nn class Model(nn.Module): def __init__(self, inputs): super (Model, self).**init**() self.layer = nn.Linear(inputs, 1) self.activation = nn.Sigmoid() def forward(self, x): x = self.laywer(x) x = self.activation(x) return x model = Model(1) print(list(model.chi.. [Pytorch] 파이토치의 구성요소, 데이터 준비 파이토치의 구성요소 torch: 메인 네임스페이스, 텐서 등의 다양한 수학 함수가 포함 torch.autograd: 자동 미분 기능을 제공하는 라이브러리 torch.nn: 신경망 구축을 위한 데이터 구조나 레이어 등의 라이브러리 torch.multiprocessing: 병럴처리 기능을 제공하는 라이브러리 torch.optim: SGD(Stochastic Gradient Descent)를 중심으로 한 파라미터 최적화 알고리즘 제공 torch.utils: 데이터 조작 등 유틸리티 기능 제공 torch.onnx: ONNX(Open Neural Network Exchange), 서로 다른 프레임워크 간의 모델을 공유할 때 사용 데이터 준비 파이토치에서는 데이터 준비를 위해 torch.utils.data의 Dat.. 이전 1 다음
