Skip to content

Commit e16e0da

Browse files
Feat: Implement DataLoader for FashionMNIST.
Changes to be committed: new file: DataLoader.ipynb
1 parent 76ceec9 commit e16e0da

File tree

1 file changed

+64
-0
lines changed

1 file changed

+64
-0
lines changed

DataLoader.ipynb

Lines changed: 64 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,64 @@
1+
{
2+
"cells": [
3+
{
4+
"cell_type": "code",
5+
"execution_count": 1,
6+
"metadata": {},
7+
"outputs": [],
8+
"source": [
9+
"import matplotlib.pyplot as plt\n",
10+
"from mxnet import gluon\n",
11+
"from mxnet.gluon.data.vision import transforms"
12+
]
13+
},
14+
{
15+
"cell_type": "code",
16+
"execution_count": 8,
17+
"metadata": {},
18+
"outputs": [],
19+
"source": [
20+
"def load_data_iter(num_workers, batch_size, shuffle=True):\n",
21+
" transformer = transforms.Compose([\n",
22+
" transforms.ToTensor()\n",
23+
" ])\n",
24+
" train = gluon.data.vision.datasets.FashionMNIST(train=True)\n",
25+
" train.transform_first(transformer)\n",
26+
" test = gluon.data.vision.datasets.FashionMNIST(train=False)\n",
27+
" test.transform_first(transformer)\n",
28+
" train_iter = gluon.data.DataLoader(train, batch_size, shuffle=shuffle, num_workers=num_workers)\n",
29+
" test_iter = gluon.data.DataLoader(train, batch_size, shuffle=shuffle, num_workers=num_workers)\n",
30+
" return train_iter, test_iter"
31+
]
32+
},
33+
{
34+
"cell_type": "code",
35+
"execution_count": 11,
36+
"metadata": {},
37+
"outputs": [],
38+
"source": [
39+
"train_iter, test_iter = load_data_iter(4, 10)"
40+
]
41+
}
42+
],
43+
"metadata": {
44+
"kernelspec": {
45+
"display_name": "Python 3",
46+
"language": "python",
47+
"name": "python3"
48+
},
49+
"language_info": {
50+
"codemirror_mode": {
51+
"name": "ipython",
52+
"version": 3
53+
},
54+
"file_extension": ".py",
55+
"mimetype": "text/x-python",
56+
"name": "python",
57+
"nbconvert_exporter": "python",
58+
"pygments_lexer": "ipython3",
59+
"version": "3.6.7"
60+
}
61+
},
62+
"nbformat": 4,
63+
"nbformat_minor": 2
64+
}

0 commit comments

Comments
 (0)