MoonNote

pytoch에서 train dataset을 validaion set 으로 split 해야할 때가 있다. MNIST 데이터셋을 예로 들면 train_dataset = datasets.MNIST('../data', train=True, download=True, transform=transform) test_dataset = datasets.MNIST('../data', train=False, transform=transform) train_dataset, val_dataset = torch.utils.data.random_split(train_dataset, [50000, 10000]) 과 같이 나누어줄 수 있다. print(len(train_dataset)) print(len(val_dataset)) p..