pytorch遇到的问题:RuntimeError: randperm is only implemented for CPU
pytorch遇到的问题:RuntimeError: randperm is only implemented for CPU
pytorch遇到的问题:RuntimeError: randperm is only implemented for CPU
由此,我们找到sample.py,第51行如下图修改
pytorch遇到的问题:RuntimeError: randperm is only implemented for CPU