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 예제를 수정해야합니다. –