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

refine lr schedule & download message (#2921)

上级 7aff0dce
...@@ -5,7 +5,7 @@ LearningRate: ...@@ -5,7 +5,7 @@ LearningRate:
schedulers: schedulers:
- !PiecewiseDecay - !PiecewiseDecay
milestones: [40, 60, 80, 100] 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 use_warmup: false
OptimizerBuilder: OptimizerBuilder:
......
...@@ -113,9 +113,9 @@ class PiecewiseDecay(object): ...@@ -113,9 +113,9 @@ class PiecewiseDecay(object):
return optimizer.lr.PiecewiseDecay(boundary, self.values) return optimizer.lr.PiecewiseDecay(boundary, self.values)
# value is computed by self.gamma # value is computed by self.gamma
if value is not None: value = value if value is not None else [base_lr]
for i in self.gamma: for i in self.gamma:
value.append(base_lr * i) value.append(base_lr * i)
return optimizer.lr.PiecewiseDecay(boundary, value) return optimizer.lr.PiecewiseDecay(boundary, value)
......
...@@ -293,16 +293,16 @@ def _dataset_exists(path, annotation, image_dir): ...@@ -293,16 +293,16 @@ def _dataset_exists(path, annotation, image_dir):
Check if user define dataset exists Check if user define dataset exists
""" """
if not osp.exists(path): if not osp.exists(path):
logger.debug("Config dataset_dir {} is not exits, " logger.warning("Config dataset_dir {} is not exits, "
"dataset config is not valid".format(path)) "dataset config is not valid".format(path))
return False return False
if annotation: if annotation:
annotation_path = osp.join(path, annotation) annotation_path = osp.join(path, annotation)
if not osp.isfile(annotation_path): if not osp.isfile(annotation_path):
logger.debug("Config annotation {} is not a " logger.warning("Config annotation {} is not a "
"file, dataset config is not " "file, dataset config is not "
"valid".format(annotation_path)) "valid".format(annotation_path))
return False return False
if image_dir: if image_dir:
image_path = osp.join(path, 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.
先完成此消息的编辑!
想要评论请 注册