将Dataset作为参数,构造一个torch.utils.data.DataLoader对象即可。 ... 自定义从数据集中取样本的策略,如果指定这个参数,那么shuffle必须为False.
確定! 回上一頁