Pytorch sampler1 Pytorch dataset(sampler) 코드를 보다가 pytorch에서 sampler를 사용하길래 간단히 아래와 같이 코드를 짜서 output을 확인해서 어떤식으로 동작하는지 찾아보았다. import random import numpy as np import torch from torch.utils.data import Dataset, RandomSampler, BatchSampler random_seed = 8138 torch.manual_seed(random_seed) torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False np.random.seed(random_seed) random.seed(random_seed) torch.cuda.manual_s.. 2022. 6. 21. 이전 1 다음