From 828959ede694f5dd3ff4361976bfcf9fe0e77ad1 Mon Sep 17 00:00:00 2001
From: p80279463 <476480381@qq.com>
Date: Fri, 30 Apr 2021 09:27:04 +0800
Subject: [PATCH] train_deepnet.py

---
 train_deepnet.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 train_deepnet.py

diff --git a/train_deepnet.py b/train_deepnet.py
new file mode 100644
index 0000000..9cdf3fb
--- /dev/null
+++ b/train_deepnet.py
@@ -0,0 +1,21 @@
+# coding: utf-8
+import sys, os
+sys.path.append(os.pardir)  # 为了导入父目录而进行的设定
+import numpy as np
+import matplotlib.pyplot as plt
+from dataset.mnist import load_mnist
+from deep_convnet import DeepConvNet
+from common.trainer import Trainer
+
+(x_train, t_train), (x_test, t_test) = load_mnist(flatten=False)
+
+network = DeepConvNet()  
+trainer = Trainer(network, x_train, t_train, x_test, t_test,
+                  epochs=20, mini_batch_size=100,
+                  optimizer='Adam', optimizer_param={'lr':0.001},
+                  evaluate_sample_num_per_epoch=1000)
+trainer.train()
+
+# 保存参数
+network.save_params("deep_convnet_params.pkl")
+print("Saved Network Parameters!")