29242f3be4bddba5143573a158f3f2e3d37ba09f,train_ssd.py,,main,#,114

Before Change

def main():
    logging.basicConfig(stream=sys.stdout, level=logging.INFO, format="%(asctime)s %(levelname)s: %(message)s")

    parser = argparse.ArgumentParser(description="Single Shot MultiBox Detector Training With PyTorch")
    parser.add_argument(
        "--config-file",

After Change

        nargs=argparse.REMAINDER,
    )
    args = parser.parse_args()
    num_gpus = int(os.environ["WORLD_SIZE"]) if "WORLD_SIZE" in os.environ else 1
    args.distributed = num_gpus > 1
    args.num_gpus = num_gpus

Instances