diff --git a/test.py b/test.py index 2bfe6b0e..89a3ea06 100644 --- a/test.py +++ b/test.py @@ -137,7 +137,7 @@ def test( mp, mr, map, mf1 = p.mean(), r.mean(), ap.mean(), f1.mean() # Print results - pf = ('%20s' + '%10.3g' * 6) # print format + pf = '%20s' + '%10.3g' * 6 # print format print(pf % ('all', seen, nt.sum(), mp, mr, map, mf1), end='\n\n') # Print results per class diff --git a/train.py b/train.py index a62612ce..7ea5b763 100644 --- a/train.py +++ b/train.py @@ -33,7 +33,7 @@ def train( img_size = 608 # initiate with maximum multi_scale size num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174 else: - torch.backends.cudnn.benchmark = True # unsuitable for multiscale + pass # torch.backends.cudnn.benchmark = True # unsuitable for multiscale # Configure run train_path = parse_data_cfg(data_cfg)['train']