Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-11-20 13:14:24 -08:00
parent bac4cc58fd
commit bd498ae776
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -173,7 +173,7 @@ def train():
model, optimizer = amp.initialize(model, optimizer, opt_level='O1', verbosity=0) model, optimizer = amp.initialize(model, optimizer, opt_level='O1', verbosity=0)
# Initialize distributed training # Initialize distributed training
if torch.cuda.device_count() > 1: if device.type != 'cpu' and torch.cuda.device_count() > 1:
dist.init_process_group(backend='nccl', # 'distributed backend' dist.init_process_group(backend='nccl', # 'distributed backend'
init_method='tcp://127.0.0.1:9999', # distributed training init method init_method='tcp://127.0.0.1:9999', # distributed training init method
world_size=1, # number of nodes for distributed training world_size=1, # number of nodes for distributed training
@ -418,6 +418,8 @@ if __name__ == '__main__':
opt.weights = last if opt.resume else opt.weights opt.weights = last if opt.resume else opt.weights
print(opt) print(opt)
device = torch_utils.select_device(opt.device, apex=mixed_precision) device = torch_utils.select_device(opt.device, apex=mixed_precision)
if device.type == 'cpu':
mixed_precision = False
# scale hyp['obj'] by img_size (evolved at 416) # scale hyp['obj'] by img_size (evolved at 416)
hyp['obj'] *= opt.img_size / 416. hyp['obj'] *= opt.img_size / 416.