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) -> None: shared_args.JobArgs.add_args(parser) TrainingDesc.add_args(parser, shared_args.maybe_get_default_hparams())
def add_args(cls, parser: argparse.ArgumentParser): defaults = shared_args.maybe_get_default_hparams() shared_args.JobArgs.add_args(parser) Branch.ArgHparams.add_args(parser) cls.BranchDesc.add_args(parser, defaults)