小编mer*_*aan的帖子

通常我正在我的电脑上进行字母和数字识别，我想将我的项目移到 Colab，但不幸的是出现了错误（你可以看到下面的错误）。经过一些调试后，我发现哪一行给了我错误。

transforms.RandomRotation(degrees=(90, -90))

下面我写了简单的抽象代码来显示这个错误。这段代码在 colab 上不起作用，但它在我自己的计算机环境中工作正常。版本 1.4.0。

import torch
import torchvision
from torchvision import datasets, transforms
import matplotlib.pyplot as plt   
    transformOpt = transforms.Compose([
            transforms.RandomRotation(degrees=(90, -90)),
            transforms.ToTensor()
        ])

    train_set = datasets.MNIST(
        root='', train=True, transform=transformOpt, download=True)
    test_set = datasets.MNIST(
        root='', train=False, transform=transformOpt, download=True)


    train_loader = torch.utils.data.DataLoader(
        dataset=train_set,
        batch_size=100,
        shuffle=True)
    test_loader = torch.utils.data.DataLoader(
        dataset=test_set,
        batch_size=100,
        shuffle=False)

    images, labels = next(iter(train_loader))
    plt.imshow(images[0].view(28, 28), cmap="gray")
    plt.show()

我在 Google Colab 上执行上面的示例代码时遇到的完整错误。

TypeError                                 Traceback (most recent call last)

<ipython-input-1-8409db422154> in <module>()
     24     shuffle=False)
     25 
---> 26 …