def add_args(parser: argparse.ArgumentParser) -> None: # Get preliminary information. defaults = shared_args.maybe_get_default_hparams() # Add the job arguments. shared_args.JobArgs.add_args(parser) lottery_parser = parser.add_argument_group( 'Lottery Ticket Hyperparameters', 'Hyperparameters that control the lottery ticket process.') LotteryRunner._add_levels_argument(lottery_parser) LotteryDesc.add_args(parser, defaults)
def add_args(parser: argparse.ArgumentParser, defaults: LotteryDesc = None): LotteryDesc.add_args(parser, defaults) BranchHparams.add_args(parser)