From c7192f64c93e48dd3b3636a0be28432a78a6f7d8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 25 Mar 2019 15:03:13 +0100 Subject: [PATCH] Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 3e983127..448f7774 100644 --- a/train.py +++ b/train.py @@ -63,7 +63,7 @@ def train( #initialize for distributed training if torch.cuda.device_count() > 1: - dist.init_process_group(backend=opt.dist_backend, init_method=opt.dist_url,world_size=opt.world_size, rank=opt.rank) + dist.init_process_group(backend=opt.backend, init_method=opt.dist_url,world_size=opt.world_size, rank=opt.rank) model = torch.nn.parallel.DistributedDataParallel(model) # Transfer learning (train only YOLO layers)