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