From 505d7e7c99619e40dd9f641c820eec2091de1d60 Mon Sep 17 00:00:00 2001 From: xiexionghang Date: Mon, 8 Jun 2020 15:30:08 +0800 Subject: [PATCH] add log --- core/utils/dataset_holder.py | 5 +++-- core/utils/envs.py | 9 +++++++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/core/utils/dataset_holder.py b/core/utils/dataset_holder.py index 66e1edb2..9b8a3d9f 100755 --- a/core/utils/dataset_holder.py +++ b/core/utils/dataset_holder.py @@ -157,8 +157,9 @@ class TimeSplitDatasetHolder(DatasetHolder): if postfix.isdigit(): if int(postfix) % node_num == node_idx: data_file_list.append(sub_file) - elif hash(sub_file_name) % node_num == node_idx: - data_file_list.append(sub_file) + else: + if hash(sub_file_name) % node_num == node_idx: + data_file_list.append(sub_file) time_window_mins = time_window_mins - self._split_interval data_time = data_time + datetime.timedelta( minutes=self._split_interval) diff --git a/core/utils/envs.py b/core/utils/envs.py index fe542979..dffc069a 100755 --- a/core/utils/envs.py +++ b/core/utils/envs.py @@ -18,6 +18,7 @@ import copy import os import socket import sys +import traceback global_envs = {} @@ -172,7 +173,9 @@ def lazy_instance_by_package(package, class_name): globals(), locals(), package.split(".")) instance = getattr(model_package, class_name) return instance - except Exception,err: + except Exception, err: + traceback.print_exc() + print('Catch Exception:%s' % str(err)) return None @@ -187,7 +190,9 @@ def lazy_instance_by_fliename(abs, class_name): globals(), locals(), package.split(".")) instance = getattr(model_package, class_name) return instance - except Exception,err: + except Exception, err: + traceback.print_exc() + print('Catch Exception:%s' % str(err)) return None -- GitLab