From 61ef6ca96897fcffe2526b5aa252d4d29b825048 Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Fri, 20 Oct 2017 15:54:03 +0200
Subject: [PATCH] flatten layers properly
---
bob/learn/tensorflow/network/SimpleCNN.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/bob/learn/tensorflow/network/SimpleCNN.py b/bob/learn/tensorflow/network/SimpleCNN.py
index bb4626bd..eb7a98a8 100644
--- a/bob/learn/tensorflow/network/SimpleCNN.py
+++ b/bob/learn/tensorflow/network/SimpleCNN.py
@@ -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
--
GitLab