質問編集履歴
1
data_number = 10000 #準備するデータの数 から3行のインデントを修正
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -32,11 +32,11 @@
|
|
|
32
32
|
|
|
33
33
|
return torch.utils.data.TensorDataset(x, y)
|
|
34
34
|
|
|
35
|
-
|
|
35
|
+
data_number = 10000 #準備するデータの数
|
|
36
36
|
|
|
37
|
-
|
|
37
|
+
batch_size = 1000 # 1つのミニバッチのデータの数
|
|
38
38
|
|
|
39
|
-
|
|
39
|
+
data_loader = torch.utils.data.DataLoader(dataset(data_number), batch_size=batch_size,shuffle=True,num_workers=2)
|
|
40
40
|
|
|
41
41
|
def imshow(img):
|
|
42
42
|
|