Bladeren bron

Assert `--image-weights` not combined with DDP (#3275)

modifyDataloader
Glenn Jocher GitHub 3 jaren geleden
bovenliggende
commit
10d56d784e
Geen bekende sleutel gevonden voor deze handtekening in de database GPG sleutel-ID: 4AEE18F83AFDEB23
1 gewijzigde bestanden met toevoegingen van 1 en 0 verwijderingen
  1. +1
    -0
      train.py

+ 1
- 0
train.py Bestand weergeven

@@ -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

Laden…
Annuleren
Opslaan