在pytorch中,使用Dataset和dataloader装载训练样本时有时需要跳过其中的一些样本。这时候只需要在Dataset对象中,将其__getitem()__函数的return为None即可。dataloader会自动跳过那些Dataset中return None的样本。
dataloader中跳过样本
最新推荐文章于 2023-12-26 19:48:09 发布
在pytorch中,使用Dataset和dataloader装载训练样本时有时需要跳过其中的一些样本。这时候只需要在Dataset对象中,将其__getitem()__函数的return为None即可。dataloader会自动跳过那些Dataset中return None的样本。