Skip to content

GitLab

"docs/source/en/optimization/mps.mdx" did not exist on "3d02c92187044393d90cc4768f9d0a0939322833"
Commit 8d849cfa authored by tink2123's avatar tink2123
Browse files

update logger

parent 97a3af3b
Show whitespace changes
Inline Side-by-side
Showing with 10 additions and 2 deletions
ppocr/data/rec/dataset_traversal.py
View file @ 8d849cfa
......@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment