From 10d56d784e7249aea0691b9c386c4828740dd9be Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 21 May 2021 14:46:42 +0200 Subject: [PATCH] Assert `--image-weights` not combined with DDP (#3275) --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 78d91fa..3aad065 100644 --- a/train.py +++ b/train.py @@ -525,6 +525,7 @@ if __name__ == '__main__': device = torch.device('cuda', opt.local_rank) dist.init_process_group(backend='nccl', init_method='env://') # distributed backend assert opt.batch_size % opt.world_size == 0, '--batch-size must be multiple of CUDA device count' + assert not opt.image_weights, '--image-weights argument is not compatible with DDP training' opt.batch_size = opt.total_batch_size // opt.world_size # Hyperparameters