2017-04-09 4 views
0

Tensorflow v1.0을 사용하여 많은 에포크로 모델을 교육하고 싶습니다. 그리고 제 생각은 모든 시대의 모든 모델을 저장하는 것입니다. 그러나 곧 나는 현재 모델이 마지막 모델을 대체 할 것임을 알았습니다. (마지막 모델은 사라질 것입니다.) 그래서 모든 모델을 얻고 하나씩 복원하는 방법을 알고 싶습니다. 나는 그것이 어렵고 좋은 해결책이 없다고 생각한다. 모든 제안에 감사드립니다!많은 단계에서 tensorflow1.0으로 모델 복원

답변

0

tf.Train.Saver().save()의 인수는 global_step입니다. documentation에서
는 :

보호기는 제공 카운터 검사 점 파일 이름을 자동으로 번호를 수 있습니다. 이를 통해 모델을 교육하는 동안 여러 단계에서 여러 검사 점을 유지할 수 있습니다. 기본적으로 Tensorflow는 지난 5 개 체크 포인트를 유지하는 것이
saver = tf.Train.Saver(...) sess = tf.Session(...) for epoch in num_epochs: ... train model... saver.save(sess, "MODEL_NAME", global_step=epoch)

참고 :

그래서 당신은 뭔가를 시도해야합니다. 모두 유지하려면 Saver을 다음 줄에 초기화해야합니다.
saver = tf.Train.Saver(max_to_keep=num_epochs)

+0

감사합니다. –

관련 문제