3
我試圖微調啓動模型,並驗證它與測試數據。但是在tensorflow給出的所有例子中,只有微調或者測試,沒有任何例子在同一個圖表和會話中進行。Tensorflow苗條列車和驗證啓動模型
基本上我想這個。
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日,以測試我的測試數據我的模型。
經過10個步驟後,培訓即將結束。要重新開始訓練,我必須從最後一個cp加載模型。但是這是非常有效的方式,我更喜歡用我的測試數據來測試我的模式。 – seleucia
爲此,您需要修改eval示例以重用訓練圖。 –