From ac7bd2e9c9d6aef3e1ddf6a5cabf1271debc2e30 Mon Sep 17 00:00:00 2001
From: pxug2blwv <mr016@qq.com>
Date: Tue, 25 Apr 2023 09:20:10 +0800
Subject: [PATCH] ADD file via upload

---
 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..cd64742
--- /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=5, 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!")