报错ValueError: sampler option is mutually exclusive with shuffle

一、问题描述

在使用torch.utils.data.DataLoader时发生报错：

ValueError: sampler option is mutually exclusive with shuffle

二、原因及解决方法

torch.utils.data.DataLoader 中，sampler 参数和 shuffle 参数不能同时指定。当 shuffle=True 时，DataLoader 会自动使用 RandomSampler；如果你同时传递了一个自定义的 sampler（或者间接传递了采样器），就会产生冲突。

所以需要在sampler和shuffle两个参数二选一