tf.Session()在pix2pix代码中冻结(CPU模式)

时间:2017-11-09 10:32:50

标签: python tensorflow

我试图在pix2pix.py代码的create_generator函数中获取隐藏的图层值(并创建PNG图像)。 我添加了这些行(在create_generator函数中):

layer_8 = tf.image.encode_png(tf.cast((tf.reshape(layers[-1], [256, 256, 3]) + 1.) * 127.5, tf.uint8))
    with tf.Session() as sess:
        sess.run(tf.global_variables_initializer())
        png_data_ = sess.run(layer_8)
        open("test_images/temp.png", 'wb').write(png_data_)

我在测试模式下运行。当我运行它时,我看到了这个警告:

2017-11-09 10:23:31.897014: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA

并且该程序不会继续运行。我正在使用CPU(没有GPU)。我究竟做错了什么?感谢。

0 个答案:

没有答案