如何了解tensorflow模型的tensorflow版本?



我只是想知道tensorflow模型(h5)文件的tensorflow版本。

我想知道tensorflow的版本号

这样的东西对你有用吗?

import h5py
import re
def get_tensorflow_version(h5):
with h5py.File(h5, 'r') as f:
metadata = f.attrs['model_config'].decode('utf-8')
version = re.search(r'"tensorflow_version":s*"([^"]*)', metadata)
if version:
return version.group(1)
else:
return "No Version"
print(get_tensorflow_version('xxx.h5'))

将xxx替换为文件名

最新更新