From 1ddea447eccb9f7182183d63e92db7c0326392e8 Mon Sep 17 00:00:00 2001
From: WuHaobo <wuhaobo1994@gmail.com>
Date: Fri, 17 Apr 2020 21:03:06 +0800
Subject: [PATCH] add sys path

---
 tools/train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/train.py b/tools/train.py
index ec6f6280..7dfbfc60 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -12,6 +12,7 @@
 #See the License for the specific language governing permissions and
 #limitations under the License.
 
+
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
@@ -19,6 +20,7 @@ from __future__ import print_function
 import argparse
 import os
 import sys
+sys.path.append(os.getcwd())
 
 import paddle
 import paddle.fluid as fluid
@@ -41,7 +43,7 @@ def parse_args():
         '-c',
         '--config',
         type=str,
-        default='configs/ResNet/ResNet18_vd.yaml',
+        default='configs/ResNet/ResNet50.yaml',
         help='config file path')
     parser.add_argument(
         '-o',
-- 
GitLab