tfcheckpoint2pytorch功能
将模型权重从TensorFlow检查点(包含*.index , *.meta和*.data-*-of-*文件的目录和压缩文件)转储至:
PyTorch二进制*.pt格式
HDF5 *.h5
NumPy *.npy和*.npz
JSON *.json
使用将TensorFlow模型从检查点导出为ONNX格式
将模型图导出到TensorBoard
相关性:不幸的是,此转换器需要安装TensorFlow(已通过v1.13.1进行了测试; v2.0可能无法正常工作)。 但是,即使通过pip安装也可以: pip3 install tensorflow 。 PyTorch,h5py,tf2onnx是可选依赖项。
示例:openseq2seq的wav2letter speech2text模型
我们将尝试将导出到ONNX。 不幸的是,tf2onnx正
2021-11-26 14:11:14
4KB
Python
1