diff --git a/train_deepnet.py b/train_deepnet.py deleted file mode 100644 index 9cdf3fb..0000000 --- a/train_deepnet.py +++ /dev/null @@ -1,21 +0,0 @@ -# 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!")