Kwong的博客

** Pytorch遇到的bug: ** RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #4 'mat2' RuntimeError: Expected object of type torch.FloatTensor but f...

解决方法来自 https://forums.fast.ai/t/databunch-from-numpy-arrays/28916 图片读入的时候是numpy array格式，是无法进行inference的，因此需要变成databunch的那种格式，官方这么说： 但是后面有人说了各更简单的方法：亲测可以。 p = untar_data(URLs.MNIST_SAMPLE) train = p/'...