본문 바로가기

ML, DL/라이브러리

Tensorflow checkpoint

출처 : https://eehoeskrap.tistory.com/343

 

[TensorFlow] .ckpt vs .pb vs .pbtxt 차이점

간단하게 말하자면 아래와 같다. ckpt 파일 모델의 변수(가중치)인 체크포인트 파일 pb 파일 모델의 변수 + 구조 (즉, 전체 그래프) 로 이루어진 바이너리 파일 pbtxt 파일 pb 파일을 읽을 수 있는 텍

eehoeskrap.tistory.com

 

train1.ckpt.data-00000-of-00001

모델의 구조를 제외한 모든 변수들을 포함, 모델 복원시 메타파일과 데이터 파일을 이용


train1.ckpt.index 

데이터파일과 메타파일을 올바르게 매핑하기 위한 내부적으로 필요한 인덱스 파일임


train1.ckpt.meta 

메타 그래프를 포함, 변수 값을 제외한 그래프의 구조를 담음

'ML, DL > 라이브러리' 카테고리의 다른 글

tensorflow, keras GPU 메모리 문제  (0) 2022.09.20
Pytorch  (0) 2022.08.02
Tensorflow, Keras  (0) 2022.08.02