From 472058363c28c89e54f7d18c8290937b5938e4e6 Mon Sep 17 00:00:00 2001 From: Gaurav Mittal Date: Sat, 4 Mar 2017 19:15:35 +0530 Subject: [PATCH] Updated network.py to support tensoflow-1.0 Please create a separate branch to support tensorflow-1.0 as previous version are used by many. --- kaffe/tensorflow/network.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/kaffe/tensorflow/network.py b/kaffe/tensorflow/network.py index 6f3b153..6351091 100644 --- a/kaffe/tensorflow/network.py +++ b/kaffe/tensorflow/network.py @@ -130,11 +130,11 @@ def conv(self, output = convolve(input, kernel) else: # Split the input into groups and then convolve each of them independently - input_groups = tf.split(3, group, input) - kernel_groups = tf.split(3, group, kernel) + input_groups = tf.split(input,group, 3) + kernel_groups = tf.split(kernel, group, 3) output_groups = [convolve(i, k) for i, k in zip(input_groups, kernel_groups)] # Concatenate the groups - output = tf.concat(3, output_groups) + output = tf.concat(output_groups, 3) # Add the biases if biased: biases = self.make_var('biases', [c_o]) @@ -177,7 +177,7 @@ def lrn(self, input, radius, alpha, beta, name, bias=1.0): @layer def concat(self, inputs, axis, name): - return tf.concat(concat_dim=axis, values=inputs, name=name) + return tf.concat(values=inputs, concat_dim=axis, name=name) @layer def add(self, inputs, name):