コード例 #1
0
ファイル: runner.py プロジェクト: sahibsin/Pruning 
 def create_from_args(args: argparse.Namespace) -> 'LotteryRunner':
     d = LotteryDesc.create_from_args(args)
     if args.weight_save_steps:
         weight_save_steps = [d.str_to_step(s) for s in args.weight_save_steps.split(',')]
     else:
         weight_save_steps = []
     return LotteryRunner(args.replicate, args.levels, LotteryDesc.create_from_args(args),
                          not args.quiet, not args.evaluate_only_at_end, weight_save_steps)
コード例 #2
0
ファイル: runner.py プロジェクト: liuguoyou/ElasticLTH 
 def create_from_args(args: argparse.Namespace) -> 'LotteryRunner':
     return LotteryRunner(args.replicate, args.levels,
                          LotteryDesc.create_from_args(args),
                          not args.quiet, not args.evaluate_only_at_end)
コード例 #3
0
ファイル: desc.py プロジェクト: liuguoyou/ElasticLTH 
 def create_from_args(cls, args: argparse.Namespace):
     return BranchDesc(LotteryDesc.create_from_args(args),
                       BranchHparams.create_from_args(args))