wolves/AI-learning
1
0
Fork 0
Code Releases Activity

250316

Browse Source
This commit is contained in:
wolves
2025-03-16 20:29:33 +08:00
parent 9a471ddedb
commit c162bd9728
2 changed files with 279 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lab/8_FC-MNIST.ipynb
View File

@@ -500,9 +500,7 @@
"# 导入数据\n",
"trainset = datasets.MNIST(root='./data',train=True,download=True,transform=transform)\n",
"trainloader = torch.utils.data.DataLoader(trainset,batch_size=64,shuffle=True)\n",
"\n",
"trainset = datasets.MNIST(root='./data',train=True,download=True,transform=transform)\n",
"testloader = torch.utils.data.DataLoader(trainset,batch_size=64,shuffle=True)\n",
"testloader = torch.utils.data.DataLoader(trainset,batch_size=64,shuffle=False)\n",
"\n",
"# 定义模型\n",
"class SimpleNet(nn.Module):\n",