如何通过google在一台机器上使用多个GPU进行物体检测来训练模型

Question

如何通过google在一台机器上使用multi gpu进行物体检测来训练模型。我设置了如下的参数：

master = ''
task = 0
num_clones = 2
clone_on_cpu = False
worker_replicas = 1
ps_tasks = 0

但我在文件“\ tensorflow \ models \ object_detection \ trainer.py”第122行的

中收到错误

_create_losses

_get_inputs(input_queue, detection_model.num_classes)

Value Error: not enough values to unpack (expected 4, got 0)

Answer 1

您需要设置batch_size = 2＆lt; - .config

Answer 2

您可以设置num_clones（要使用的GPU的数量）和ps_tasks（参数服务器）参数

python3 train.py --logtostderr \ --pipeline_config_path=model/ssd_inception_v2.config \ --train_dir=model/train \ --num_clones=2 --ps_tasks=1