質問編集履歴
1
コードの追加
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -13,5 +13,12 @@
|
|
|
13
13
|
transform=transforms.ToTensor(),
|
|
14
14
|
download=True,
|
|
15
15
|
)
|
|
16
|
+
|
|
17
|
+
train_loader = torch.utils.data.DataLoader(
|
|
18
|
+
dataset=train_dataset,
|
|
19
|
+
batch_size=BATCH_SIZE, #256
|
|
20
|
+
shuffle=True,
|
|
21
|
+
num_workers=0
|
|
22
|
+
)
|
|
16
23
|
```
|
|
17
24
|
しかし，0,6の2種だけのデータセットはどのように用意すればいいのかわからず見当がつかない状況です．もしご存知の方がいましたら，どうか教えていただけると幸いでございます．どうぞよろしくお願いいたします．
|