Skip to content
Snippets Groups Projects
Commit 61ef6ca9 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

flatten layers properly

parent 64efc538
Branches
Tags
1 merge request!23Change default feature
Pipeline #
This commit is part of merge request !23. Comments created here will be created in the context of that merge request.
......@@ -47,8 +47,8 @@ def architecture(input_layer, mode=tf.estimator.ModeKeys.TRAIN,
data_format=data_format)
# Flatten tensor into a batch of vectors
dim = tf.reduce_prod(tf.shape(pool2)[1:])
pool2_flat = tf.reshape(pool2, [-1, dim])
# TODO: use tf.layers.flatten in tensorflow 1.4 above
pool2_flat = tf.contrib.layers.flatten(pool2)
# Dense Layer
# Densely connected layer with 1024 neurons
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment