首页 > 解决方案 > TensorFlow 目标检测多 GPU 参数

问题描述

对于大多数模型,为了使用 3 个 GPU,我可以--num_clones=3 --ps_tasks=1在运行训练时设置。但是,对于一些模型,这不起作用。还有其他参数,例如worker_replicas, num_ps_tasks, num_replicas.

有人能解释一下这些参数背后的含义,并举例说明如何将它们设置为使用具有 N 个 GPU 的单台机器吗?

顺便说一句,我要运行的模型是ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config

标签: tensorflowparametersobject-detection

解决方案


推荐阅读