def main()

in pai-python-sdk/training/pytorch_ddp/train_src/train_multinode.py [0:0]

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

def main(
    save_every: int,
    total_epochs: int,
    batch_size: int,
    output_model_path: str,
    checkpoint_path: str,