ddu_dirty_mnist 源码

DDU的肮脏MNIST 您再也不想将MNIST用于OOD或AL。 安装 pip install ddu_dirty_mnist 如何使用 安装后,您将获得Dirty-MNIST训练或测试仪,就像在PyTorch中使用MNIST一样。 # gpu import ddu_dirty_mnist dirty_mnist_train = ddu_dirty_mnist . DirtyMNIST ( "." , train = True , download = True , device = "cuda" ) dirty_mnist_test = ddu_dirty_mnist . DirtyMNIST ( "." , train = False , download = True , device = "cuda" ) len ( dirty_mnist_train ),