cc:184 Key generator/g_d1_bn/beta not found in checkpoint

最新推荐文章于 2022-03-22 13:13:42 发布

Tzu-Heng

最新推荐文章于 2022-03-22 13:13:42 发布

阅读量696

点赞数

文章标签： tensorflow dl ml zi2zi

本文链接：https://blog.csdn.net/weixin_43809009/article/details/88971176

版权

在执行zi2zi项目时遇到错误：Key generator/g_d1_bn/beta not found in checkpoint。使用的是TensorFlow，并在GPU上运行。尝试加载模型时发生此错误。问题可能涉及模型检查点的不匹配或缺失变量。

摘要由CSDN通过智能技术生成

目前我正在執行github上的專案pcgreat/zi2zi的專案
執行過程中發生錯誤
log如下

這是我的test.sh文件
PYTHONPATH=. python infer.py --model_dir=experiments/checkpoint/experiment_0
–batch_size=32
–source_obj=experiments/data/val.obj
–embedding_ids=0
–save_dir=save_dir/
以下是執行過程
(py36) sjshyu@sjshyu-titan:~/Desktop/360315$ sh test.sh
2019-03-29 14:10:03.478788: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
2019-03-29 14:10:03.660390: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1356] Found device 0 with properties:
name: GeForce GTX TITAN X major: 5 minor: 2 memoryClockRate(GHz): 1.076
pciBusID: 0000:02:00.0
totalMemory: 11.92GiB freeMemory: 11.17GiB
2019-03-29 14:10:03.792179: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1356] Found device 1 with properties:
name: GeForce GTX TITAN X major: 5 minor: 2 memoryClockRate(GHz): 1.076
pciBusID: 0000:81:00.0
totalMemory: 11.92GiB freeMemory: 11.80GiB
2019-03-29 14:10:03.792283: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1435] Adding visible gpu devices: 0, 1
2019-03-29 14:10:04.332448: I tensorflow/core/common_runtime/gpu/gpu_device.cc:923] Device interconnect StreamExecutor with strength 1 edge matrix:
2019-03-29 14:10:04.332508: I tensorflow/core/common_runtime/gpu/gpu_device.cc:929] 0 1
2019-03-29 14:10:04.332519: I tensorflow/core/common_runtime/gpu/gpu_device.cc:942] 0: N N
2019-03-29 14:10:04.332524: I tensorflow/core/common_runtime/gpu/gpu_device.cc:942] 1: N N
2019-03-29 14:10:04.333112: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1053] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 10816 MB memory) -> physical GPU (device: