D:\Python\Anaconda\envs\pyG\python.exe D:/Python/pycharm/pycharm_code/Python_PyTorch/第三章/PyTorch神经网络工具箱.py
Dataset MNIST
Number of datapoints:60000
Root location: D:\Python\python_dataset\torch
Split: Train
StandardTransform
Transform: Compose(
ToTensor()
Normalize(mean=[0.5], std=[0.5]))
Dataset MNIST
Number of datapoints:10000
Root location: D:\Python\python_dataset\torch
Split: Test
StandardTransform
Transform: Compose(
ToTensor()
Normalize(mean=[0.5], std=[0.5]))<enumerateobject at 0x000001E3A80BFF98>
torch.Size([128,1,28,28])
torch.Size([128])
torch.Size([28,28])0.009000000000000001
epoch:0, Train Loss:0.2251, Train Acc:0.9360, Test Loss:0.0913, Test Acc:0.9729
epoch:1, Train Loss:0.0872, Train Acc:0.9738, Test Loss:0.0740, Test Acc:0.9775
epoch:2, Train Loss:0.0563, Train Acc:0.9831, Test Loss:0.0646, Test Acc:0.9812
epoch:3, Train Loss:0.0447, Train Acc:0.9860, Test Loss:0.0643, Test Acc:0.9797
epoch:4, Train Loss:0.0334, Train Acc:0.9899, Test Loss:0.0635, Test Acc:0.97990.008100000000000001
epoch:5, Train Loss:0.0239, Train Acc:0.9931, Test Loss:0.0558, Test Acc:0.9835
epoch:6, Train Loss:0.0205, Train Acc:0.9941, Test Loss:0.0538, Test Acc:0.9842
epoch:7, Train Loss:0.0164, Train Acc:0.9952, Test Loss:0.0576, Test Acc:0.9828
epoch:8, Train Loss:0.0148, Train Acc:0.9958, Test Loss:0.0559, Test Acc:0.9837
epoch:9, Train Loss:0.0114, Train Acc:0.9971, Test Loss:0.0546, Test Acc:0.98360.007290000000000001
epoch:10, Train Loss:0.0092, Train Acc:0.9974, Test Loss:0.0555, Test Acc:0.9832
epoch:11, Train Loss:0.0072, Train Acc:0.9983, Test Loss:0.0527, Test Acc:0.9840
epoch:12, Train Loss:0.0064, Train Acc:0.9985, Test Loss:0.0509, Test Acc:0.9847
epoch:13, Train Loss:0.0061, Train Acc:0.9987, Test Loss:0.0540, Test Acc:0.9844
epoch:14, Train Loss:0.0047, Train Acc:0.9990, Test Loss:0.0577, Test Acc:0.98350.006561000000000002
epoch:15, Train Loss:0.0056, Train Acc:0.9985, Test Loss:0.0522, Test Acc:0.9839
epoch:16, Train Loss:0.0044, Train Acc:0.9989, Test Loss:0.0544, Test Acc:0.9840
epoch:17, Train Loss:0.0038, Train Acc:0.9992, Test Loss:0.0549, Test Acc:0.9840
epoch:18, Train Loss:0.0040, Train Acc:0.9991, Test Loss:0.0541, Test Acc:0.9850
epoch:19, Train Loss:0.0031, Train Acc:0.9994, Test Loss:0.0547, Test Acc:0.9852