Open chatterboy opened 5 years ago
TensorFlow에서 모델을 학습시키면, 그래프 정의를 포함하는 graph_def 파일과 그래프 내 학습된 가중치를 포함하는 .ckpt 파일이 생성된다. 위의 두 파일을 .onnx 파일로 export하기 위해서 TensorFlow에서는 Graph Freezing을 제공한다. 즉, 위의 두 파일을 하나의 파일로 만들어주는데, Variable 노드에 있는 텐서의 값을 Const로 만들어 .ckpt 파일에 있는 값을 저장한다.
graph_def
.ckpt
.onnx
export
Graph Freezing
Variable
Const
TensorFlow to ONNX
TensorFlow에서 모델을 학습시키면, 그래프 정의를 포함하는
graph_def
파일과 그래프 내 학습된 가중치를 포함하는.ckpt
파일이 생성된다. 위의 두 파일을.onnx
파일로export
하기 위해서 TensorFlow에서는Graph Freezing
을 제공한다. 즉, 위의 두 파일을 하나의 파일로 만들어주는데,Variable
노드에 있는 텐서의 값을Const
로 만들어.ckpt
파일에 있는 값을 저장한다.