| 11 | |
| 12 | |
| 13 | def add_encoding_training_options(parser): |
| 14 | group = parser.add_argument_group("encoding") |
| 15 | group.add_argument("--data_path", type=str, help="path to source data") |
| 16 | group.add_argument("--enc_batch_size", type=int, default=65536, help="batch size") |
| 17 | group.add_argument("--fm_reso", type=int, default=128, help="feature map resolution") |
| 18 | group.add_argument("--sdf_renorm", type=int, default=0, help="renormalize sdf values to [-1, 1]") |
| 19 | group.add_argument("--data_type", type=str, default="sdftex", choices=["sdf", "sdftex", "sdfpbr"], help="data type") |
| 20 | |
| 21 | group.add_argument("--enc_net_type", type=str, default="skip", help="network type") |
| 22 | group.add_argument("-fdg", "--fdim_geo", type=int, default=4, help="geometry feature dimension") |
| 23 | group.add_argument("-fdt", "--fdim_tex", type=int, default=8, help="texture feature dimension") |
| 24 | group.add_argument("-fdup", "--fdim_up", type=int, default=64, help="conv feature dimension") |
| 25 | group.add_argument("-hd", "--hidden_dim", type=int, default=256, help="mlp hidden dimension") |
| 26 | group.add_argument("-nh", "--n_hidden_layers", type=int, default=4, help="mlp hidden layers") |
| 27 | |
| 28 | group.add_argument("--enc_n_iters", type=int, default=25000, help="total number of epochs to train") |
| 29 | group.add_argument("--enc_lr", type=float, default=5e-3, help="initial learning rate") |
| 30 | group.add_argument("--enc_lr_decay", type=float, default=0.1, help="initial learning rate") |
| 31 | group.add_argument("--enc_lr_split", type=float, default=0.2, help="") |
| 32 | group.add_argument("--vol_ratio", type=float, default=0.1, help="vol points ratio") |
| 33 | group.add_argument("--tex_threshold_ratio", type=float, default=0.999, help="tex threshold ratio") |
| 34 | group.add_argument("--tex_weight", type=float, default=1.0, help="tex weight") |
| 35 | group.add_argument("--sdf_loss", type=str, default="weightedl1", choices=["l1", "weightedl1"]) |
| 36 | group.add_argument("--tex_loss", type=str, default="l1", choices=["l1", "l2", "huber"]) |
| 37 | |
| 38 | |
| 39 | def add_diffusion_training_options(parser): |