Пример #1
0
    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)
Пример #2
0
 def add_args(parser: argparse.ArgumentParser,
              defaults: LotteryDesc = None):
     LotteryDesc.add_args(parser, defaults)
     BranchHparams.add_args(parser)