Hi, I wonder if I am getting the following error because the code has changed from 0.4 to 1 or maybe because server error (due to unresolved packages)? I have no problem running it on my machine but after loading pytorch on server I get this error (which has PyTorch 1).
[jalal@scc-c12 src]$ module load torchvision
ModuleCmd_Load.c(208):ERROR:105: Unable to locate a modulefile for 'torchvision'
NameError Traceback (most recent call last)
<ipython-input-3-0b0ce4a30f28> in <module>()
29 # should I use transforms.RandomRotation(10) instead?
30 data_transforms = {
---> 31 'train': transforms.Compose([
32 transforms.RandomResizedCrop(224),
33 transforms.RandomHorizontalFlip(),
NameError: name 'transforms' is not defined
This is the version I have on my own machine:
torch.__version__
'0.4.1.post2'