updates
This commit is contained in:
		
							parent
							
								
									37a07a44a1
								
							
						
					
					
						commit
						01e11aee04
					
				
							
								
								
									
										2
									
								
								train.py
								
								
								
								
							
							
						
						
									
										2
									
								
								train.py
								
								
								
								
							| 
						 | 
				
			
			@ -76,7 +76,7 @@ def train(
 | 
			
		|||
 | 
			
		||||
    if multi_scale:
 | 
			
		||||
        img_size = round((img_size / 32) * 1.5) * 32  # initiate with maximum multi_scale size
 | 
			
		||||
        # opt.num_workers = 0  # bug https://github.com/ultralytics/yolov3/issues/174
 | 
			
		||||
        opt.num_workers = 0  # bug https://github.com/ultralytics/yolov3/issues/174
 | 
			
		||||
    else:
 | 
			
		||||
        torch.backends.cudnn.benchmark = True  # unsuitable for multiscale
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue