search

514flowey / JDet-cobb

The Jittor Implementation of COBB
Apache License 2.0
10 stars 0 forks source link

multi-gpu training #8

Closed YXB-NKU closed 1 month ago

YXB-NKU commented 1 month ago

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 JITTOR_HOME=build/ python tools/run_net.py \ --config-file configs/oriented_rcnn_r50_fpn_1x_dota_with_flip.py --task=train

when i use this command to train , i found only gpu0 is working,how can i use multi-gpu to train my config