运行startup_program失败
收藏
images = fluid.layers.data(name='images', shape=[3,240,240], dtype='float32')#如果把[3,240,240]改成[240,240,3]可以往下运行但数据格式就不对了
label = fluid.layers.data(name='label', shape=[1], dtype='float32')
predict = cnn(images)
cost = fluid.layers.cross_entropy(input=predict, label=label)
avg_cost = fluid.layers.mean(cost)
label = fluid.layers.data(name='label', shape=[1], dtype='float32')
acc = fluid.layers.accuracy(input=predict, label=label)
#定义优化器
optimizer = fluid.optimizer.AdamOptimizer(learning_rate=0.001)
optimizer.minimize(avg_cost)
print('优化')
use_cuda = False
place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
exe = fluid.Executor(place)
# 初始化执行器
print('star')
exe.run(fluid.default_startup_program()) #运行到这就开始不往下运行了,也不报错
0
收藏
请登录后评论
exe.run(fluid.default_startup_program()) #运行到这就开始不往下运行了,也不报错
这里确实不会报错,因为你只是搭建了网络但是没有feed数据,所以机器肯定不会给你反馈。一般来说接下来是用类似
的语句将你的数据feed到网络中进行训练,default_startup_program()只是初始化网络而已~
要运行得把数据feed进去