torch.utils.data.random_split(dataset, lengths)

37

>>> random_split(range(10), [3, 7], generator=torch.Generator().manual_seed(42))

Comments

Submit
0 Comments