torch.no_grad()

[Pytorch] 파이토치 TIL 모음

Pytorch를 사용하면서 알게된 내용들, 정확히 이해하려는 코드들을 모아놓습니다. Updated 21.10.07 1. TensorDataset의 사용 torch.tensor -> TensorDataset -> DataLoader 의 과정으로 깔끔한 코드를 작성한다. from torch.utils.data import (DataLoader, TensorDataset) TensorDataset은 ds = TensorDataset(x,y) 뿐만 아니라 여러개의 데이터를 넣을 수도 있다. # 보통 사용 ds = TensorDataset(x,y) # 여러개 묶음 사용 train_features = TensorDatast(input, input_lengths, label, left_feature, right_fe..