Edit by browser

2020-07-13 14:05:01 +08:00 · 2020-07-13 14:05:01 +08:00 · 65a856f918
parent 45273fb117
commit 65a856f918
1 changed files with 3 additions and 3 deletions
--- a/MindSpore/src/step2/step2.py
+++ b/MindSpore/src/step2/step2.py
@ -22,11 +22,11 @@ if __name__ == "__main__":
    # define the optimizer
    # 请在此添加代码完成本关任务
    #********** Begin *********#
-    ## 提示：补全优化器的定义
-
+    ## 提示：补全优化器的定义，使用 Momentum 优化器

    #********** End **********#
-
+    config_ck = CheckpointConfig(save_checkpoint_steps=1875, keep_checkpoint_max=10)
+    
    # save the network model and parameters for subsequence fine-tuning
    ckpoint_cb = ModelCheckpoint(prefix="checkpoint_lenet", config=config_ck)
    # group layers into an object with training and evaluation features