Skip to content

Commit c0e100a

Browse files
merge develop
2 parents dfd0a0d + 7284e3f commit c0e100a

File tree

3 files changed

+36
-8
lines changed

3 files changed

+36
-8
lines changed

paddleclas.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,7 @@ def print_info():
340340
first_width = 30
341341
second_width = total_width - first_width if total_width > 50 else 10
342342
except OSError:
343+
total_width = 100
343344
second_width = 100
344345
for series in IMN_MODEL_SERIES:
345346
names = textwrap.fill(
@@ -452,7 +453,9 @@ class PaddleClas(object):
452453
"""PaddleClas.
453454
"""
454455

455-
print_info()
456+
if not os.environ.get('ppcls', False):
457+
os.environ.setdefault('ppcls', 'True')
458+
print_info()
456459

457460
def __init__(self,
458461
model_name: str=None,

ppcls/utils/logger.py

Lines changed: 29 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
import datetime
16+
import logging
1517
import os
1618
import sys
1719

18-
import logging
19-
import datetime
2020
import paddle.distributed as dist
2121

2222
_logger = None
@@ -39,22 +39,45 @@ def init_logger(name='ppcls', log_file=None, log_level=logging.INFO):
3939
logging.Logger: The expected logger.
4040
"""
4141
global _logger
42-
assert _logger is None, "logger should not be initialized twice or more."
43-
_logger = logging.getLogger(name)
42+
43+
# solve mutiple init issue when using paddleclas.py and engin.engin
44+
init_flag = False
45+
if _logger is None:
46+
_logger = logging.getLogger(name)
47+
init_flag = True
4448

4549
formatter = logging.Formatter(
4650
'[%(asctime)s] %(name)s %(levelname)s: %(message)s',
4751
datefmt="%Y/%m/%d %H:%M:%S")
4852

4953
stream_handler = logging.StreamHandler(stream=sys.stdout)
5054
stream_handler.setFormatter(formatter)
51-
_logger.addHandler(stream_handler)
55+
stream_handler._name = 'stream_handler'
56+
57+
# add stream_handler when _logger dose not contain stream_handler
58+
for i, h in enumerate(_logger.handlers):
59+
if h.get_name() == stream_handler.get_name():
60+
break
61+
if i == len(_logger.handlers) - 1:
62+
_logger.addHandler(stream_handler)
63+
if init_flag:
64+
_logger.addHandler(stream_handler)
65+
5266
if log_file is not None and dist.get_rank() == 0:
5367
log_file_folder = os.path.split(log_file)[0]
5468
os.makedirs(log_file_folder, exist_ok=True)
5569
file_handler = logging.FileHandler(log_file, 'a')
5670
file_handler.setFormatter(formatter)
57-
_logger.addHandler(file_handler)
71+
file_handler._name = 'file_handler'
72+
73+
# add file_handler when _logger dose not contain same file_handler
74+
for i, h in enumerate(_logger.handlers):
75+
if h.get_name() == file_handler.get_name() and \
76+
h.baseFilename == file_handler.baseFilename:
77+
break
78+
if i == len(_logger.handlers) - 1:
79+
_logger.addHandler(file_handler)
80+
5881
if dist.get_rank() == 0:
5982
_logger.setLevel(log_level)
6083
else:

test_tipc/prepare.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,9 @@ if [[ ${MODE} = "cpp_infer" ]]; then
9898

9999
if [[ $cpp_type == "cls" ]]; then
100100
eval "wget -nc $cls_inference_url"
101-
tar xf "${model_name}_infer.tar"
101+
tar_name=$(func_get_url_file_name "$cls_inference_url")
102+
model_dir=${tar_name%.*}
103+
eval "tar xf ${tar_name}"
102104

103105
cd dataset
104106
rm -rf ILSVRC2012

0 commit comments

Comments
 (0)