Evaluation arguments.
(parser)
| 53 | |
| 54 | |
| 55 | def add_evaluation_args(parser): |
| 56 | """Evaluation arguments.""" |
| 57 | |
| 58 | group = parser.add_argument_group('validation', 'validation configurations') |
| 59 | |
| 60 | group.add_argument('--eval-batch-size', type=int, default=None, |
| 61 | help='Data Loader batch size for evaluation datasets.' |
| 62 | 'Defaults to `--batch-size`') |
| 63 | group.add_argument('--eval-iters', type=int, default=100, |
| 64 | help='number of iterations to run for evaluation' |
| 65 | 'validation/test for') |
| 66 | group.add_argument('--eval-interval', type=int, default=None, |
| 67 | help='interval between running evaluation on validation set') |
| 68 | group.add_argument('--strict-eval', action='store_true', |
| 69 | help='won\'t enlarge or randomly map eval-data, and eval full eval-data.') |
| 70 | |
| 71 | group.add_argument('--sampling_video_size', type=list, default=[480, 720]) |
| 72 | group.add_argument('--n_prediction_round', type=int, default=1) |
| 73 | group.add_argument('--use_ema', type=bool, default=False) |
| 74 | group.add_argument('--apply_traj', type=bool, default=False) |
| 75 | |
| 76 | group.add_argument('--save_recon', type=bool, default=False) # * Save memory |
| 77 | group.add_argument('--save_gt', type=bool, default=True) |
| 78 | |
| 79 | group.add_argument('--concat_recon_for_demo', type=bool, default=False) # * Save memory |
| 80 | group.add_argument('--concat_gt_for_demo', type=bool, default=False) |
| 81 | |
| 82 | group.add_argument('--n_cond_frames', type=int, default=3) # * DEFAULT: 3 |
| 83 | |
| 84 | return parser |
| 85 | |
| 86 | |
| 87 | def get_args(args_list=None, parser=None): |