diff --git a/python/paddle/fluid/trainer_desc.py b/python/paddle/fluid/trainer_desc.py index 4a27ea3fd8871d3b3a5c62582c7cc65b2144c3c1..f323439c55071436989a363b7f8d1fc840b65fc1 100644 --- a/python/paddle/fluid/trainer_desc.py +++ b/python/paddle/fluid/trainer_desc.py @@ -14,7 +14,7 @@ """Defination of trainers.""" import sys -from os import path +import os __all__ = ['TrainerDesc', 'MultiTrainer', 'DistMultiTrainer', 'PipelineTrainer'] @@ -32,9 +32,16 @@ class TrainerDesc(object): ''' # Workaround for relative import in protobuf under python3 # TODO: should be fixed - cur_path = path.dirname(__file__) + cur_path = os.path.dirname(__file__) sys.path.append(cur_path) sys.path.append(cur_path + "/proto") + + # Just for fix random UT fail, will be deleted soon + print("cur_path: ", cur_path) + print("sys.path: ", sys.path) + if os.path.isdir(cur_path + "/proto"): + print("[listdir]") + print(os.listdir(cur_path + "/proto")) from proto import trainer_desc_pb2 self.proto_desc = trainer_desc_pb2.TrainerDesc() import multiprocessing as mp