찰떡개발자 코딩

Tensorflow2.x를 사용하지만 1점대를 사용하며 공부하고 있어서 이점 참고하길 바란다.

1). ChechPoint란?

• Tensorflow 로 학습시킨 딥러닝 모델을 저장하는 방법중 하나로 Checkpoint 을 이용하는 방법이 있다.
• Checkpoint 는 학습된 모델의 Variable 값을 저장하는 파일이다.

Checkpoint는 개발자가 딥러닝 모델을 구현해놓은것을 저장해 놓고 로드해서 사용하는것으로 생각하면 된다.

2). CheckPoint 저장방법

#Checkpoint 저장
saver = tf.train.Saver()
saver.save(sess,"D:\\DATA\\test.cpkt")

#Checkpoint 불러와서 저장한다.
save_path = "D:\\DATA\\baechooe.cpkt"
saver.restore(sess, save_path)

3). 오류

ValueError: The passed save_path is not a valid checkpoint: D:\DATA\baechooe.cpkt

=> 해결

본 오류는 checkpoint를 잘못 지정했다고 말하는것이다.

checkpoint를 저장할때 지정한 경로의 해당 checkpoint를 불러와주면 해결된다.