wangtao
/
MindSpore-Model-Development
forked from mindspore-Ecosystem/MindSpore-Model-Development
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Edit by browser

This commit is contained in:
陈亚云 2020-07-13 13:55:06 +08:00
parent 1109ba19a3
commit 45273fb117
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
MindSpore/src/step2/step2.py
View File

@ -12,19 +12,20 @@ if __name__ == "__main__":
epoch_size = 1 epoch_size = 1
mnist_path = "./MNIST_Data" mnist_path = "./MNIST_Data"
# 请在此添加代码完成本关任务 # 请在此添加代码完成本关任务
# **********Begin*********# #********** Begin *********#
##提示：补全损失函数的定义 ## 提示：补全损失函数的定义
# **********End**********# #********** End **********#
repeat_size = epoch_size repeat_size = epoch_size
# create the network # create the network
network = LeNet5() network = LeNet5()
# define the optimizer # define the optimizer
# 请在此添加代码完成本关任务 # 请在此添加代码完成本关任务
# **********Begin*********# #********** Begin *********#
##提示：补全优化器的定义 ## 提示：补全优化器的定义
# **********End**********#
#********** End **********#
# save the network model and parameters for subsequence fine-tuning # save the network model and parameters for subsequence fine-tuning
ckpoint_cb = ModelCheckpoint(prefix="checkpoint_lenet", config=config_ck) ckpoint_cb = ModelCheckpoint(prefix="checkpoint_lenet", config=config_ck)