활연개랑

[tensorflow] checking weights in checkpoint 본문

딥러닝

[tensorflow] checking weights in checkpoint

승해tmdhey 2023. 1. 13. 14:18
반응형
import tensorflow as tf
from tensorflow.python.training import py_checkpoint_reader

checkpoint_path = f'/cp-{123:0>4}.ckpt'
reader     = py_checkpoint_reader.NewCheckpointReader(checkpoint_path)
dtype_map  = reader.get_variable_to_dtype_map()
shape_map  = reader.get_variable_to_shape_map()
state_dict = { v: reader.get_tensor(v) for v in shape_map}


with open('weight_check.txt', 'w') as file:
    file.write(str(state_dict))

'딥러닝' 카테고리의 다른 글

calculation of atrous convolution receptive field  (0) 2023.01.16
Confusion matrix / Segmentation quality metrics  (0) 2023.01.13
Function call stack: train_function -> train_function -> train_function (Input to reshape is a tensor with 15360 values, but the requested shape has 7680)  (0) 2022.09.14
[Linux] zip 압축해제 ERROR : extra bytes at beginning or within zipfile  (0) 2022.06.15
[tensorflow] error해결 : train_function -> train_function -> train_function -> train_function 오류  (0) 2022.06.10
'딥러닝' Related Articles more