溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

Tensorflow1.0之后模型文件、權(quán)重數(shù)值的讀取方式是什么

發(fā)布時間：2021-11-05 10:01:40 來源：億速云 閱讀：167 作者：iii 欄目：編程語言

本篇內(nèi)容主要講解“Tensorflow1.0之后模型文件、權(quán)重數(shù)值的讀取方式是什么”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學(xué)習(xí)“Tensorflow1.0之后模型文件、權(quán)重數(shù)值的讀取方式是什么”吧!

保存的文件有4個：

checkpoint

model-parameters.bin-46000.data-00000-of-00001

model-parameters.bin-46000.index

model-parameters.bin-46000.meta

讀取代碼：

from tensorflow.python import pywrap_tensorflowwith tf.Graph().as_default(): with tf.Session() as sess: ckpt = tf.train.get_checkpoint_state('./checkpoint') if ckpt and ckpt.model_checkpoint_path:  reader = pywrap_tensorflow.NewCheckpointReader('./checkpoint/model-parameters.bin-46000')  all_variables = reader.get_variable_to_shape_map()  w1 = reader.get_tensor("conv1/weight")  print(w1.shape)  print(w1)  print(all_variables) else:  print('No checkpoint file found')

到此，相信大家對“Tensorflow1.0之后模型文件、權(quán)重數(shù)值的讀取方式是什么”有了更深的了解，不妨來實際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進入相關(guān)頻道進行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手