Skip to content

Commit 2898c58

Browse files
fthoelesoumith
authored andcommitted
Use variable instead of magic number in data_parallel_tutorial.py (pytorch#244)
1 parent e83f94b commit 2898c58

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/blitz/data_parallel_tutorial.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def __getitem__(self, index):
7979
def __len__(self):
8080
return self.len
8181

82-
rand_loader = DataLoader(dataset=RandomDataset(input_size, 100),
82+
rand_loader = DataLoader(dataset=RandomDataset(input_size, data_size),
8383
batch_size=batch_size, shuffle=True)
8484

8585

0 commit comments

Comments
 (0)