Type Error on Resize Method

[leacornelio]

I'm trying to train on ImageNet but the following error is showing:

Traceback (most recent call last): File "imagenet.py", line 121, in main('mobile',args.batch_size, args.root, args.n_epochs) File "imagenet.py", line 110, in main trainer.loop(n_epochs, train_loader, test_loader) File "imagenet.py", line 62, in loop self.train(train_data) File "imagenet.py", line 51, in train loss, correct = self._loop(data_loader) File "imagenet.py", line 33, in _loop for data, target in tqdm(data_loader): File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/tqdm/_tqdm.py", line 937, in iter for obj in iterable: File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 336, in next return self._process_next_batch(batch) File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 357, in _process_next_batch raise batch.exc_type(batch.exc_msg) TypeError: Traceback (most recent call last): File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 106, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torchvision/datasets/folder.py", line 103, in getitem sample = self.transform(sample) File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torchvision/transforms/transforms.py", line 49, in call img = t(img) File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torchvision/transforms/transforms.py", line 546, in call return F.resized_crop(img, i, j, h, w, self.size, self.interpolation) File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torchvision/transforms/functional.py", line 331, in resized_crop img = resize(img, size, interpolation) File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/torchvision/transforms/functional.py", line 206, in resize return img.resize(size[::-1], interpolation) File "/home/ubuntu/anaconda3/envs/frenv/lib/python2.7/site-packages/PIL/Image.py", line 1765, in resize return self._new(self.im.resize(size, resample, box)) TypeError: an integer is required