INNER CODE UNIT · Python
CLI
Linketic/CityGaussian · internal/cli.py:28
class CLI(LightningCLI):
def add_arguments_to_parser(self, parser: LightningArgumentParser) -> None:
parser.add_argument("--max_steps", "--iterations", "--iteration", "--steps", "--step", type=Optional[int], default=None)
parser.add_argument("--max_epochs", "--epochs", "--epoch", type=Optional[int], default=None)
parser.add_argument("--name", "-n", type=Optional[str], default=None,
help="the training result output path will be 'output/name'")
parser.add_argument("--version", "-v", type=Optional[str], default=None,
help="the training result output path will be 'output/name/version'")
# TODO: add max_steps to save_iterations, but need to compatible with --max_steps < 0 & --max_epochs > 0
parser.add_argument("--save_iterations", type=List[int], default=[7_000, 30_000])
parser.add_argument("--logger", type=str, default="tensorboard")
parser.add_argument("--project", type=str, default="Gaussian-Splatting", help="WanDB project name")
parser.add_argument("--output", type=str, default=os.path.join(
os.path.dirname(os.path.dirname(__file__)),
"outputs",
), help="the base directory of the output")
parser.add_argument("--float32_matmul_precision", "-f", type=Optional[Literal["medium", "high", "highest"]], default=None)
parser.add_argument("--viewer", action="store_true", default=False)