this arg:
deploy_config = model_deploy.DeploymentConfig(
num_clones=FLAGS.num_clones,
clone_on_cpu=FLAGS.clone_on_cpu,
replica_id=0,
num_replicas=1,
num_ps_tasks=0)
and
config = tf.ConfigProto(log_device_placement=False,
gpu_options=gpu_options,
allow_soft_placement=True)
it is work !!
if i set multiGPU here will be raise an error:
if num_replicas > 1:
if num_ps_tasks < 1:
raise ValueError('When using replicas num_ps_tasks must be positive')
this arg: deploy_config = model_deploy.DeploymentConfig( num_clones=FLAGS.num_clones, clone_on_cpu=FLAGS.clone_on_cpu, replica_id=0, num_replicas=1, num_ps_tasks=0) and config = tf.ConfigProto(log_device_placement=False, gpu_options=gpu_options, allow_soft_placement=True) it is work !!
num_replicas=4, num_ps_tasks=4 is error ..
Originally posted by @zsz02 in https://github.com/balancap/SSD-Tensorflow/issues/58#issuecomment-297600787