質問編集履歴
1
data_number = 10000 #準備するデータの数 から3行のインデントを修正
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -15,9 +15,9 @@
|
|
|
15
15
|
x = torch.FloatTensor(x)
|
|
16
16
|
y = torch.FloatTensor(y)
|
|
17
17
|
return torch.utils.data.TensorDataset(x, y)
|
|
18
|
-
|
|
18
|
+
data_number = 10000 #準備するデータの数
|
|
19
|
-
|
|
19
|
+
batch_size = 1000 # 1つのミニバッチのデータの数
|
|
20
|
-
|
|
20
|
+
data_loader = torch.utils.data.DataLoader(dataset(data_number), batch_size=batch_size,shuffle=True,num_workers=2)
|
|
21
21
|
def imshow(img):
|
|
22
22
|
img = img / 2 + 0.5 # 正規化を戻す
|
|
23
23
|
npimg = img.numpy()
|