|
|
import argparse |
|
|
|
|
|
def get_args_parser(): |
|
|
parser = argparse.ArgumentParser(description='Optimal Transport AutoEncoder training for AIST', |
|
|
add_help=True, |
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
|
|
|
|
|
|
|
|
parser.add_argument('--dataname', type=str, default='t2m_272', help='dataset directory') |
|
|
parser.add_argument('--batch-size', default=128, type=int, help='batch size') |
|
|
parser.add_argument('--window-size', type=int, default=64, help='training motion length') |
|
|
|
|
|
|
|
|
parser.add_argument('--total-iter', default=2000000, type=int, help='number of total iterations to run') |
|
|
parser.add_argument('--warm-up-iter', default=1000, type=int, help='number of total iterations for warmup') |
|
|
parser.add_argument('--lr', default=5e-5, type=float, help='max learning rate') |
|
|
parser.add_argument('--lr-scheduler', default=[50000, 400000], nargs="+", type=int, help="learning rate schedule (iterations)") |
|
|
parser.add_argument('--gamma', default=0.05, type=float, help="learning rate decay") |
|
|
|
|
|
parser.add_argument('--weight-decay', default=0.0, type=float, help='weight decay') |
|
|
|
|
|
|
|
|
parser.add_argument("--down-t", type=int, default=2, help="downsampling rate") |
|
|
parser.add_argument("--stride-t", type=int, default=2, help="stride size") |
|
|
parser.add_argument("--depth", type=int, default=3, help="depth of the network") |
|
|
parser.add_argument("--dilation-growth-rate", type=int, default=3, help="dilation growth rate") |
|
|
|
|
|
|
|
|
parser.add_argument("--resume-pth", type=str, default=None, help='resume pth for causal TAE') |
|
|
|
|
|
|
|
|
|
|
|
parser.add_argument('--out-dir', type=str, default='output/', help='output directory') |
|
|
parser.add_argument('--results-dir', type=str, default='visual_results/', help='output directory') |
|
|
parser.add_argument('--visual-name', type=str, default='vis', help='output directory') |
|
|
parser.add_argument('--exp-name', type=str, default='exp', help='name of the experiment, will create a file inside out-dir') |
|
|
parser.add_argument('--latent_dir', type=str, default='t2m_latents/', help='latent directory') |
|
|
|
|
|
parser.add_argument('--print-iter', default=200, type=int, help='print frequency') |
|
|
parser.add_argument('--eval-iter', default=20000, type=int, help='evaluation frequency') |
|
|
parser.add_argument('--seed', default=123, type=int, help='seed for initializing training.') |
|
|
|
|
|
parser.add_argument('--vis-gt', action='store_true', help='whether visualize GT motions') |
|
|
parser.add_argument('--nb-vis', default=20, type=int, help='nb of visualizations') |
|
|
parser.add_argument('--root_loss', default=7.0, type=float, help='root loss') |
|
|
parser.add_argument('--latent_dim', default=16, type=int, help='latent dimension') |
|
|
parser.add_argument('--hidden_size', default=1024, type=int, help='hidden size') |
|
|
parser.add_argument('--nb_joints', default=22, type=int, help='number of joints') |
|
|
parser.add_argument('--num_gpus', default=1, type=int, help='number of GPUs') |
|
|
|
|
|
return parser.parse_args() |
