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日,以測試我的測試數據我的模型。

回答

-1

是否用number_of_steps = 10調用列車,然後使用評估代碼工作?

+0

經過10個步驟後,培訓即將結束。要重新開始訓練,我必須從最後一個cp加載模型。但是這是非常有效的方式,我更喜歡用我的測試數據來測試我的模式。 – seleucia

+0

爲此,您需要修改eval示例以重用訓練圖。 –