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