From ba1a0a9fed8194513c5a3be30cfa3e04780960e9 Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Thu, 7 May 2020 08:55:15 +0000
Subject: [PATCH] fix license year, output dir

---
 examples/handwritten_number_recognition/mnist.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/examples/handwritten_number_recognition/mnist.py b/examples/handwritten_number_recognition/mnist.py
index 3407849..36db96c 100644
--- a/examples/handwritten_number_recognition/mnist.py
+++ b/examples/handwritten_number_recognition/mnist.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
+# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -60,7 +60,7 @@ def main():
               val_dataset,
               epochs=FLAGS.epoch,
               batch_size=FLAGS.batch_size,
-              save_dir='mnist_checkpoint')
+              save_dir=FLAGS.output_dir)
 
 
 if __name__ == '__main__':
@@ -81,7 +81,10 @@ if __name__ == '__main__':
     parser.add_argument(
         "-b", "--batch_size", default=128, type=int, help="batch size")
     parser.add_argument(
-        "--output-dir", type=str, default='output', help="checkpoint save dir")
+        "--output-dir",
+        type=str,
+        default='mnist_checkpoint',
+        help="checkpoint save dir")
     parser.add_argument(
         "-r",
         "--resume",
-- 
GitLab