NameError: name 'transforms' is not defined, while using google colab

Question

train = datasets.MNIST("", train=True, download=True,transform = transforms.Compose([transforms.ToTensor()]))
test = datasets.MNIST("", train=False, download=True,transform = transforms.Compose([transforms.ToTensor()]))`

After executing this on colab notebook, I am getting this error:

Traceback (most recent call last)
<ipython-input-6-b81aa6cf1cbe> in <module>()
----> 1 train = datasets.MNIST("", train=True, download=True,transform = transforms.Compose([transforms.ToTensor()]))
      2 test = datasets.MNIST("", train=False, download=True,transform = transforms.Compose([transforms.ToTensor()]))

NameError: name 'transforms' is not defined*

Why do you think it should be defined? Did you import it?

Mark
– Mark

2020-05-17 18:43:04 +00:00
Commented May 17, 2020 at 18:43 — Mark
– Mark, Commented May 17, 2020 at 18:43

SimonR · Accepted Answer · 2020-05-17 18:44:10Z

12

I'm guessing from the context that you're using Pytorch, in which case you need to make sure you have:

from torchvision import transforms

In your imports. By the looks of things you have already imported datasets from the same library.

answered May 17, 2020 at 18:44

SimonR

1,8241 gold badge7 silver badges12 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

NameError: name 'transforms' is not defined, while using google colab

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related