()
| 14 | |
| 15 | |
| 16 | def parse_args(): |
| 17 | parser = configargparse.ArgParser() |
| 18 | parser.add('-c', '--config', required=True, is_config_file=True, help='Config file path') |
| 19 | parser.add("--name", type=str, default="main") |
| 20 | parser.add("--train_data_path", action="append") |
| 21 | parser.add("--val_data_path", action="append") |
| 22 | parser.add("--test_data_path", action="append") |
| 23 | parser.add("--model_save_path", required=True) |
| 24 | parser.add("--pose_representation", type=str, default='3d_vec') |
| 25 | parser.add("--mean_dir_vec", action="append", type=float, nargs='*') |
| 26 | parser.add("--mean_pose", action="append", type=float, nargs='*') |
| 27 | parser.add("--random_seed", type=int, default=-1) |
| 28 | parser.add("--save_result_video", type=str2bool, default=True) |
| 29 | |
| 30 | # word embedding |
| 31 | parser.add("--wordembed_path", type=str, default=None) |
| 32 | parser.add("--wordembed_dim", type=int, default=100) |
| 33 | parser.add("--freeze_wordembed", type=str2bool, default=False) |
| 34 | |
| 35 | # model |
| 36 | parser.add("--model", type=str, required=True) |
| 37 | parser.add("--epochs", type=int, default=10) |
| 38 | parser.add("--batch_size", type=int, default=50) |
| 39 | parser.add("--dropout_prob", type=float, default=0.3) |
| 40 | parser.add("--n_layers", type=int, default=2) |
| 41 | parser.add("--hidden_size", type=int, default=200) |
| 42 | parser.add("--z_type", type=str, default='none') |
| 43 | parser.add("--input_context", type=str, default='both') |
| 44 | |
| 45 | # dataset |
| 46 | parser.add("--motion_resampling_framerate", type=int, default=24) |
| 47 | parser.add("--n_poses", type=int, default=50) |
| 48 | parser.add("--n_pre_poses", type=int, default=5) |
| 49 | parser.add("--subdivision_stride", type=int, default=5) |
| 50 | parser.add("--loader_workers", type=int, default=16) |
| 51 | |
| 52 | # GAN parameter |
| 53 | parser.add("--GAN_noise_size", type=int, default=0) |
| 54 | |
| 55 | # training |
| 56 | parser.add("--learning_rate", type=float, default=0.001) |
| 57 | parser.add("--discriminator_lr_weight", type=float, default=0.2) |
| 58 | parser.add("--loss_regression_weight", type=float, default=50) |
| 59 | parser.add("--loss_gan_weight", type=float, default=1.0) |
| 60 | parser.add("--loss_kld_weight", type=float, default=0.1) |
| 61 | parser.add("--loss_reg_weight", type=float, default=0.01) |
| 62 | parser.add("--loss_warmup", type=int, default=-1) |
| 63 | |
| 64 | # eval |
| 65 | parser.add("--eval_net_path", type=str, default='') |
| 66 | |
| 67 | args = parser.parse_args() |
| 68 | return args |
no outgoing calls
no test coverage detected