- 博客(2)
- 收藏
- 关注
原创 keras(tensorflow后端)给模型建立单独的graph
因为要在一个项目中同时导入多个模型,需要给keras模型新建个graph,但是按照tensorflow新建graph的方式一直类似的报错:ValueError: Tensor Tensor("Sigmoid_2:0", shape=(?, 17), dtype=float32) is not an element of this graph.原因大概是多线程、分布式环境下,恢复Model时的Tens
2018-01-18 11:06:56 4525 5
原创 在一个项目中导入多个不同tensorflow模型
刚开始直接采用调用一个模型的方法: (1)定义网络 (2)新建sess:sess = tf.Session(config=config) (3)定义saver:saver = tf.train.Saver() (4)导入权重:saver.restore(sess, xxx) 但是,如果在一个项目中同时导入多个模型,会报错,应该是graph冲突,所以需要给每个模型单独新建graph:g1 =
2018-01-18 10:58:23 8608 7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人