목록텐서플로우 (1)

머신러닝과 기술적 분석

[Tensorflow] Checkpoint file에 저장되어있는 Variable Tensor를 알아보는 방법

TensorFlow는 학습한 모델을 2개의 file로 나누어서 저장한다. meta file 에는 code 로 구현한 graph가 저장이 되고 (.meta file) ckeckpoint file 에는 training 과정에서 결정된 Variable Tensor의 value가 저장된다. (.ckpt file) 따라서 두 개의 file이 있으면 학습한 network model을 load 할수 가 있다. 그런데 TensorFlow Model Zoo 처럼 ckeckpoint file 만을 제공하는 경우도 있고, ckeckpoint 에 저장되어 있는 variable tensor의 scope/name을 알고 싶을 때가 있다. 이럴 때는 tf.contrib.framework.list_variables() 를 사용하면 ..
Tensorflow 2017. 8. 19. 16:13
Prev 1 Next