Aymeric Damien 9 éve
szülő
commit
0094bf0b0f
1 módosított fájl, 3 hozzáadás és 3 törlés
  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):