浏览代码

Correct tf types

Aymeric Damien 9 年之前
父节点
当前提交
0094bf0b0f
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      examples/3 - Neural Networks/alexnet.py

+ 3 - 3
examples/3 - Neural Networks/alexnet.py

@@ -25,9 +25,9 @@ n_classes = 10 # MNIST total classes (0-9 digits)
 dropout = 0.8 # Dropout, probability to keep units
 
 # tf Graph input
-x = tf.placeholder(tf.types.float32, [None, n_input])
-y = tf.placeholder(tf.types.float32, [None, n_classes])
-keep_prob = tf.placeholder(tf.types.float32) # dropout (keep probability)
+x = tf.placeholder(tf.float32, [None, n_input])
+y = tf.placeholder(tf.float32, [None, n_classes])
+keep_prob = tf.placeholder(tf.float32) # dropout (keep probability)
 
 # Create AlexNet model
 def conv2d(name, l_input, w, b):