search

hukaixuan19970627 / DOTA_devkit_YOLO

Trans DOTA OBB format(poly format) to YOLO format.
198 stars 39 forks source link

Error running the training file；Multi-distributed training #11

Closed Alice-cmd666 closed 2 years ago

Alice-cmd666 commented 2 years ago

I have two video cards. I changed the parameters of the corresponding video card during training, but the code of the training file runs incorrectly. subprocess.CalledProcessError: Command '['/root/miniconda3/envs/yolov5/bin/python', '-u', 'train.py', '--local_rank=1', '--sync-bn']' returned non-zero exit status 1.

hukaixuan19970627 commented 2 years ago 
$   python -m torch.distributed.launch --nproc_per_node 2 train.py --sync-bn --device 0,1

Is this command cause the error?