Commit d2cde908 authored by yhcao6's avatar yhcao6
Browse files

change the interface of build_sampler and build_assigner, replace default_args with **kwargs

parent 6a363603
...@@ -3,23 +3,23 @@ import mmcv ...@@ -3,23 +3,23 @@ import mmcv
from . import assigners, samplers from . import assigners, samplers
def build_assigner(cfg, default_args=None): def build_assigner(cfg, **kwargs):
if isinstance(cfg, assigners.BaseAssigner): if isinstance(cfg, assigners.BaseAssigner):
return cfg return cfg
elif isinstance(cfg, dict): elif isinstance(cfg, dict):
return mmcv.runner.obj_from_dict( return mmcv.runner.obj_from_dict(
cfg, assigners, default_args=default_args) cfg, assigners, default_args=kwargs)
else: else:
raise TypeError('Invalid type {} for building a sampler'.format( raise TypeError('Invalid type {} for building a sampler'.format(
type(cfg))) type(cfg)))
def build_sampler(cfg, default_args=None): def build_sampler(cfg, **kwargs):
if isinstance(cfg, samplers.BaseSampler): if isinstance(cfg, samplers.BaseSampler):
return cfg return cfg
elif isinstance(cfg, dict): elif isinstance(cfg, dict):
return mmcv.runner.obj_from_dict( return mmcv.runner.obj_from_dict(
cfg, samplers, default_args=default_args) cfg, samplers, default_args=kwargs)
else: else:
raise TypeError('Invalid type {} for building a sampler'.format( raise TypeError('Invalid type {} for building a sampler'.format(
type(cfg))) type(cfg)))
......
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