Skip to content

GitLab

Commit f3d5e85d authored by vfdev's avatar vfdev Committed by Francisco Massa
Browse files

Update fakedata.py (#581) 

Target `float` -> `long`
parent 152c2b3a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datasets/fakedata.py
View file @ f3d5e85d
...@@ -40,7 +40,7 @@ class FakeData(data.Dataset): ...@@ -40,7 +40,7 @@ class FakeData(data.Dataset):
rng_state = torch.get_rng_state() rng_state = torch.get_rng_state()
torch.manual_seed(index + self.random_offset) torch.manual_seed(index + self.random_offset)
img = torch.randn(*self.image_size) img = torch.randn(*self.image_size)
target = torch.Tensor(1).random_(0, self.num_classes)[0] target = torch.randint(0, self.num_classes, size=(1,), dtype=torch.long)[0]
torch.set_rng_state(rng_state) torch.set_rng_state(rng_state)
# convert to PIL Image # convert to PIL Image
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment