通過設定export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7指定8卡GPU訓練。 通過設定MASK_ON選擇Faster RCNN和Mask RCNN模型。 可選引數見:. python train.
確定! 回上一頁