Slide 17
Slide 17 text
学習の様子
● n_hidden = 10、n_hidden = 128 など、値を変えて実行
$ make train_gpu
:
2021-12-22 08:13:44,860 - INFO - model = MNISTModel(
(linear_relu_stack): Sequential(
(0): Linear(in_features=784, out_features=10, bias=True)
(1): ReLU()
(2): Linear(in_features=10, out_features=10, bias=True)
)
)
100%|████████████████████████████████████████████████| 938/938 [00:07<00:00, 119.43it/s]
2021-12-22 08:13:52,718 - INFO - Epoch [1/5] train loss: 0.64784, acc: 0.80790
2021-12-22 08:13:53,915 - INFO - Epoch [1/5] valid loss: 0.38300, acc: 0.89070
100%|████████████████████████████████████████████████| 938/938 [00:07<00:00, 118.71it/s]
2021-12-22 08:14:01,819 - INFO - Epoch [2/5] train loss: 0.37280, acc: 0.89393
2021-12-22 08:14:03,024 - INFO - Epoch [2/5] valid loss: 0.35724, acc: 0.89690
100%|████████████████████████████████████████████████| 938/938 [00:07<00:00, 119.12it/s]
2021-12-22 08:14:10,901 - INFO - Epoch [3/5] train loss: 0.34369, acc: 0.90180
2021-12-22 08:14:12,099 - INFO - Epoch [3/5] valid loss: 0.33862, acc: 0.90290
100%|████████████████████████████████████████████████| 938/938 [00:07<00:00, 118.21it/s]
2021-12-22 08:14:20,037 - INFO - Epoch [4/5] train loss: 0.32955, acc: 0.90513
2021-12-22 08:14:21,240 - INFO - Epoch [4/5] valid loss: 0.33413, acc: 0.90330
100%|████████████████████████████████████████████████| 938/938 [00:07<00:00, 119.53it/s]
2021-12-22 08:14:29,090 - INFO - Epoch [5/5] train loss: 0.31951, acc: 0.90883
2021-12-22 08:14:30,267 - INFO - Epoch [5/5] valid loss: 0.31779, acc: 0.91120
2021-12-22 08:14:30,268 - INFO - export onnx model
: