error log

error log

(wind_torch_2022) bim@bim-PowerEdge-R730:~/Pytorch_Project/yolov5_train_warning$ python train_20220602.py --batch-size 1 --epochs 300 --data ./data/myvoc.yaml --cfg ./models/yolov5x.yaml --workers 0   --device 0,1
Traceback (most recent call last):
  File "train_20220602.py", line 496, in <module>
    device = select_device(opt.device, batch_size=opt.batch_size)
  File "/home/bim/Pytorch_Project/yolov5_train_warning/utils/torch_utils.py", line 58, in select_device
    assert batch_size % n == 0, f'batch-size {batch_size} not multiple of GPU count {n}'
AssertionError: batch-size 1 not multiple of GPU count 2
(wind_torch_2022) bim@bim-PowerEdge-R730:~/Pytorch_Project/yolov5_train_warning$ 
(wind_torch_2022) bim@bim-PowerEdge-R730:~/Pytorch_Project/yolov5_train_warning$ 

 

 

batch-size 2

python train_20220602.py --batch-size 2  --epochs 300 --data ./data/myvoc.yaml --cfg ./models/yolov5x.yaml --workers 0   --device 0,1

 

 

##############