def parse_args()

in dataflux_pytorch/benchmark/checkpointing/multinode/train.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--save_only", action="store_true", default=False)
    parser.add_argument("--load_only", action="store_true", default=False)
    parser.add_argument(
        '--strategy',
        choices=[DF_FSDP_STRATEGY, FSSPEC_FSDP_STRATEGY, FSDP_STRATEGY],
        default=DF_FSDP_STRATEGY)
    parser.add_argument("--distributed_filesystem",
                        action="store_true",
                        default=False)
    return parser.parse_args()