| 531 | |
| 532 | |
| 533 | def parse_args(): |
| 534 | |
| 535 | parser = argparse.ArgumentParser() |
| 536 | parser.add_argument('--dataroot', default='ShapeNetCore.v2.PC15k/') |
| 537 | parser.add_argument('--category', default='chair') |
| 538 | |
| 539 | parser.add_argument('--batch_size', type=int, default=50, help='input batch size') |
| 540 | parser.add_argument('--workers', type=int, default=16, help='workers') |
| 541 | parser.add_argument('--niter', type=int, default=10000, help='number of epochs to train for') |
| 542 | |
| 543 | parser.add_argument('--generate',default=True) |
| 544 | parser.add_argument('--eval_gen', default=True) |
| 545 | |
| 546 | parser.add_argument('--nc', default=3) |
| 547 | parser.add_argument('--npoints', default=2048) |
| 548 | '''model''' |
| 549 | parser.add_argument('--beta_start', default=0.0001) |
| 550 | parser.add_argument('--beta_end', default=0.02) |
| 551 | parser.add_argument('--schedule_type', default='linear') |
| 552 | parser.add_argument('--time_num', default=1000) |
| 553 | |
| 554 | #params |
| 555 | parser.add_argument('--attention', default=True) |
| 556 | parser.add_argument('--dropout', default=0.1) |
| 557 | parser.add_argument('--embed_dim', type=int, default=64) |
| 558 | parser.add_argument('--loss_type', default='mse') |
| 559 | parser.add_argument('--model_mean_type', default='eps') |
| 560 | parser.add_argument('--model_var_type', default='fixedsmall') |
| 561 | |
| 562 | |
| 563 | parser.add_argument('--model', default='',required=True, help="path to model (to continue training)") |
| 564 | |
| 565 | '''eval''' |
| 566 | |
| 567 | parser.add_argument('--eval_path', |
| 568 | default='') |
| 569 | |
| 570 | parser.add_argument('--manualSeed', default=42, type=int, help='random seed') |
| 571 | |
| 572 | parser.add_argument('--gpu', type=int, default=0, metavar='S', help='gpu id (default: 0)') |
| 573 | |
| 574 | opt = parser.parse_args() |
| 575 | |
| 576 | if torch.cuda.is_available(): |
| 577 | opt.cuda = True |
| 578 | else: |
| 579 | opt.cuda = False |
| 580 | |
| 581 | return opt |
| 582 | if __name__ == '__main__': |
| 583 | opt = parse_args() |
| 584 | set_seed(opt) |