Question

我正在尝试微调初始模型，并使用测试数据对其进行验证。但是在tensorflow slime网页上给出的所有示例都只是微调或测试，没有任何例子可以在同一个图和会话中同时进行。

基本上我想要这个。

with tf.Graph().as_default():
 image, image_raw,  label,image_name, label_name = dut.distorted_inputs(params,is_training=is_training)
 test_image, test_image_raw,  test_label,test_image_name, test_label_name = dut.distorted_inputs(params,is_training=False)

 # I'm creating as it is suggested at github slim page:
 logits, _ =inception.inception_v2(image, num_classes=N, is_training=True)
 tf.get_variable_scope().reuse_variables()
 logits_tes, _ =inception.inception_v2(test_image, num_classes=N, is_training=Test)
 err=tf.sub(logits, label)
 losses = tf.reduce_mean(tf.reduce_sum(tf.square(err)))
 # total_loss = model_loss+losses
 total_loss = losses+slim.losses.get_total_loss()
 test_err=tf.sub(test_logits, test_label)
 test_loss= tf.reduce_mean(tf.reduce_sum(tf.square(test_err)))

 optimizer = tf.train.AdamOptimizer(learning_rate=0.001)
 train_op = slim.learning.create_train_op(total_loss, optimizer)

 final_loss = slim.learning.train(
     train_op,
     logdir=params["cp_file"],
     init_fn=ut.get_init_fn(slim,params),
     number_of_steps=2,
     summary_writer=summary_writer
 )

此代码失败可以看出，我没有单独循环来调用我的测试模型，我想在每个第10批测试数据上测试我的模型。

Answer 1

使用number_of_steps = 10调用train然后使用评估代码工作吗？

Tensorflow细长列车和验证初始模型

1 个答案: