如何在CPU上运行CUDNN兼容GRUCELL训练的权重

时间:2019-01-14 22:04:36

标签: tensorflow deep-learning artificial-intelligence recurrent-neural-network cudnn

我已经在Titan Titan上使用“ CUDNN兼容GRUCELL”训练了GRU,但是我试图在CPU上运行保存的权重,但是模型的结果有问题。考虑到我在GPU上的测试结果是98%,我该如何在CPU上运行图形。由于管理员的原因,我无法在经过训练的GPU上运行图形。

race_5000_graph = tf.Graph()
race_5000_session = tf.Session(graph=race_5000_graph)

saver = tf.train.import_meta_graph(modelCheckpointMeta)
saver.restore(race_5000_session,modelCheckpoint)

feed={x:bn, sl: [input_length]*256, kp:1.0}

race_percents = race_5000_session.run(fs, feed_dict=feed)  

上面的代码是片段,它们初始化并运行了我的预训练图。

0 个答案:

没有答案