feeded_var_names多变量报错
梦意凉心更冷s 发布于2019-09 浏览:1838 回复:3
0
收藏

今天做一个多输入的模型时,用save_inference_model保存模型,feeded_var_names包含多个输入变量名时,报Value Error

大概意思是block里没有'chars',但是我输入数据命名没有问题

有没有大佬知道是为什么呀

求救~~~~

 

收藏
点赞
0
个赞
共3条回复 最后由lemonadeqk回复于2021-02
#4lemonadeqk回复于2021-02

你好!我也遇到跟你类似的问题,我的也出现这个报错,看到你说“网络结构写错了”,能具体说一下,你最终是怎么解决的吗?

 

0
#3梦意凉心更冷s回复于2019-10

找到错误了,网络结构写错了

 

0
#2梦意凉心更冷s回复于2019-09
# 定义输入
chars = fluid.layers.data(name='chars', shape=[1], dtype='int64', lod_level=1)
words = fluid.layers.data(name='words', shape=[1], dtype='int64', lod_level=1)
label = fluid.layers.data(name='label', shape=[1], dtype='int64')

...

# 定义数据映射器
feeder = fluid.DataFeeder(place=place, feed_list=[words,chars,label])

...
# 保存预测模型
if not os.path.exists(model_save_dir): 
    os.makedirs(model_save_dir) 
fluid.io.save_inference_model(model_save_dir, 
                            feeded_var_names=['words','chars'], 
                            target_vars=[model], 
                            executor=exe0)
print('训练模型保存完成!') 

0
TOP
切换版块