Skip to content

Commit becf45d

Browse files
Update image_classification_tutorial.ipynb
1 parent 256e234 commit becf45d

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

examples/image_classification_tutorial.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,6 +409,7 @@
409409
" mapping=mapping,\n",
410410
" mode=\"test\")\n",
411411
"\n",
412+
"# If you have lower amount of processing power set num_workers to 0\n"
412413
"train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=32, shuffle=True, num_workers=2)\n",
413414
"val_loader = torch.utils.data.DataLoader(valid_dataset, batch_size=1, shuffle=False, num_workers=2)\n",
414415
"test_loader = torch.utils.data.DataLoader(test_dataset,batch_size=1, shuffle=False, num_workers=2)"
@@ -933,4 +934,4 @@
933934
},
934935
"nbformat": 4,
935936
"nbformat_minor": 1
936-
}
937+
}

0 commit comments

Comments
 (0)