def parse_block_cfg(block_op, out_channels, stride=1, repeat=1, *args):
assert all(isinstance(x, dict) for x in args), f"{args}"
cfg = {"out_channels": out_channels, "stride": stride}
[hp.update_dict(cfg, x) for x in args]
ret = {"block_op": block_op, "block_cfg": cfg, "repeat": repeat}
return ret
def add_basic_args(self, **kwargs):
""" args that will be passed to all primitives, they could be
overrided by primitive parameters
"""
hp.update_dict(self.basic_args, kwargs)