tensorflow 모델 복원 예외 설정 방법

Question

내가하려는 것은 오픈 소스 텐서 흐름 모델에 몇 가지 기능을 추가하는 것입니다.

그 일을하는 동안

, 나는, 내가 모델이 내 컴퓨터에 훈련하기 어렵 기 때문에 검사 ​​점을 복원하기 위해 노력하고있어 동시에 tf.Vriable (안 tf.get_variable)

를 사용하여 새로운 변수를 추가했습니다.

내가 체크 포인트에서 매개 변수를 복원하기 위해 노력하고있어, NotFoundError가 발생

문제

이있다. 오류 메시지는 다음과 같습니다.

NotFoundError (see above for traceback): Key INITIAL_VAR not found in checkpoint 
[[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/cpu:0"](_arg_save/Const_0_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]] 

특정 변수에 대한 chekpoint에서 복원 방지하기 위해 예외를 설정하는 방법은 없나요?

읽어 주셔서 감사합니다.

Answer 1

tf.train.Saver()의 var_list 인수를 사용하면되고 싶었습니다.

나는 내가 답을 찾은 것 같아이

variable_list = tf.global_variables() 
variable_list.remove(variable_that_I_dont_want_to_restore) 
saver = tf.train.Saver(var_list=variable_list) 
saver.restore(sess,checkpoint)