1.修改配置(faster_rcnn_resnet101.config):
batch_size: 2
2.添加运行参数:
--num_clones=2 --ps_tasks=1
CUDA_VISIBLE_DEVICES=0,1 python object_detection/train.py \
--logtostderr \
--pipeline_config_path=gen/data/faster_rcnn_resnet101.config \
--train_dir=gen/train \
--num_clones=2 --ps_tasks=1
参考
https://blog.csdn.net/jidebingfeng/article/details/81557434
https://github.com/tensorflow/models/issues/1972