小姜贼菜

[07/01 12:01:29 d2.utils.events]: eta: 2 days, 5:45:52 iter: 1799 total_loss: 42.75 loss_x: 3.227 loss_y: 3.196 loss_w: 1.471 loss_h: 1.835 loss_conf: 26.53 loss_cls: 6.332 time: 0.2771 data_time: 0.0038 lr: 0.00018833 max_mem:...

python -m torch.distributed.run --nproc_per_node 2 train.py --data coco.yaml --cfg ./configs/backbone/yolov5_res2net50.yaml --batch-size 16 --epoch 300 --device 0,1 --weight ''

同问

cuda version pytorch version?