Skip to content

Commit 761e369

Browse files
authored
Merge pull request pytorch#175 from eminenceG/master
Change 'dataloders' to 'dataloaders'
2 parents 140eb2a + 38b6544 commit 761e369

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

beginner_source/transfer_learning_tutorial.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@
9191
image_datasets = {x: datasets.ImageFolder(os.path.join(data_dir, x),
9292
data_transforms[x])
9393
for x in ['train', 'val']}
94-
dataloders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=4,
94+
dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=4,
9595
shuffle=True, num_workers=4)
9696
for x in ['train', 'val']}
9797
dataset_sizes = {x: len(image_datasets[x]) for x in ['train', 'val']}
@@ -119,7 +119,7 @@ def imshow(inp, title=None):
119119

120120

121121
# Get a batch of training data
122-
inputs, classes = next(iter(dataloders['train']))
122+
inputs, classes = next(iter(dataloaders['train']))
123123

124124
# Make a grid from batch
125125
out = torchvision.utils.make_grid(inputs)
@@ -163,7 +163,7 @@ def train_model(model, criterion, optimizer, scheduler, num_epochs=25):
163163
running_corrects = 0
164164

165165
# Iterate over data.
166-
for data in dataloders[phase]:
166+
for data in dataloaders[phase]:
167167
# get the inputs
168168
inputs, labels = data
169169

@@ -225,7 +225,7 @@ def visualize_model(model, num_images=6):
225225
images_so_far = 0
226226
fig = plt.figure()
227227

228-
for i, data in enumerate(dataloders['val']):
228+
for i, data in enumerate(dataloaders['val']):
229229
inputs, labels = data
230230
if use_gpu:
231231
inputs, labels = Variable(inputs.cuda()), Variable(labels.cuda())

0 commit comments

Comments
 (0)