parent
1d91ac5926
commit
ab6e7725d5
@ -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!")
|
|
Loading…
Reference in new issue