未验证 提交 75d29d16 编写于 作者: W wangguanzhong 提交者: GitHub

refine lr schedule & download message (#2921)

上级 7aff0dce
......@@ -5,7 +5,7 @@ LearningRate:
schedulers:
- !PiecewiseDecay
milestones: [40, 60, 80, 100]
values: [0.001, 0.0005, 0.00025, 0.0001, 0.00001]
gamma: [0.5, 0.5, 0.4, 0.1]
use_warmup: false
OptimizerBuilder:
......
......@@ -113,9 +113,9 @@ class PiecewiseDecay(object):
return optimizer.lr.PiecewiseDecay(boundary, self.values)
# value is computed by self.gamma
if value is not None:
for i in self.gamma:
value.append(base_lr * i)
value = value if value is not None else [base_lr]
for i in self.gamma:
value.append(base_lr * i)
return optimizer.lr.PiecewiseDecay(boundary, value)
......
......@@ -293,16 +293,16 @@ def _dataset_exists(path, annotation, image_dir):
Check if user define dataset exists
"""
if not osp.exists(path):
logger.debug("Config dataset_dir {} is not exits, "
"dataset config is not valid".format(path))
logger.warning("Config dataset_dir {} is not exits, "
"dataset config is not valid".format(path))
return False
if annotation:
annotation_path = osp.join(path, annotation)
if not osp.isfile(annotation_path):
logger.debug("Config annotation {} is not a "
"file, dataset config is not "
"valid".format(annotation_path))
logger.warning("Config annotation {} is not a "
"file, dataset config is not "
"valid".format(annotation_path))
return False
if image_dir:
image_path = osp.join(path, image_dir)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册