Parcourir la source

Correct tf types

Aymeric Damien il y a 9 ans
Parent
commit
0094bf0b0f
1 fichiers modifiés avec 3 ajouts et 3 suppressions
  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):