You can use the torch.utils.data.Subset class which takes in input a dataset and a set of indices and selects only the elements ...
確定! 回上一頁