Function get_args

train_code/train_wan_motion.py:137–160

()

def get_args():
    parser = argparse.ArgumentParser(description="Simple example of a ControlNet training script.")
    parser.add_argument(
                            "--config_path",
                            type = str,
                            default = "config/train_wan_motion.yaml",
                            # required = True,
                            help = "Path to the config.",
                        )
    parser.add_argument(
                            "--use_8BitAdam",
                            type = bool,
                            default = False,
                            help = "Whether we use the 8BitAdam.",
                        )
    parser.add_argument(
                            "--debug",
                            type = bool,
                            default = False,
                            help = "Whether we are in the debug model: don't use torch.compile",
                        )

    args = parser.parse_args()
    return args