Slim :Cannot assign a device for operation 'InceptionV3/Predictions/Softmax'錯誤
阿新 • • 發佈:2018-11-11
########################### # Kicks off the training. # ########################### session_config = tf.ConfigProto(allow_soft_placement=True) slim.learning.train( train_tensor, logdir=FLAGS.train_dir, master=FLAGS.master, is_chief=(FLAGS.task == 0), init_fn=_get_init_fn(), summary_op=summary_op, number_of_steps=FLAGS.max_number_of_steps, log_every_n_steps=FLAGS.log_every_n_steps, save_summaries_secs=FLAGS.save_summaries_secs, save_interval_secs=FLAGS.save_interval_secs, sync_optimizer=optimizer if FLAGS.sync_replicas else None, session_config=session_config, )