提交 8d849cfa 编写于 作者: T tink2123

update logger

上级 97a3af3b
......@@ -46,7 +46,11 @@ class LMDBReader(object):
if "tps" in params:
self.ues_tps = True
if "distort" in params:
self.use_distort = params['distort']
self.use_distort = params['distort'] and params['use_gpu']
if not params['use_gpu']:
logger.info(
"Distort operation can only support in GPU. Distort will be set to False."
)
if params['mode'] == 'train':
self.batch_size = params['train_batch_size_per_card']
self.drop_last = True
......@@ -189,7 +193,11 @@ class SimpleReader(object):
if "tps" in params:
self.use_tps = True
if "distort" in params:
self.use_distort = params['distort']
self.use_distort = params['distort'] and params['use_gpu']
if not params['use_gpu']:
logger.info(
"Distort operation can only support in GPU.Distort will be set to False."
)
if params['mode'] == 'train':
self.batch_size = params['train_batch_size_per_card']
self.drop_last = True
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册