Add logs to debug random fail (#23731)

revert-23830-2.0-beta
hutuxian 5 years ago committed by GitHub
parent f26f7c3603
commit 6f7077db25
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save