openai/improved-gan

ValueError: squeeze_dims[1] not in [-2,2). for 'Squeeze_1' (op: 'Squeeze') with input shapes: [?,2048].

youkaichao opened this issue · 1 comments

on the calculation of inception score, after pool3 = sess.graph.get_tensor_by_name('pool_3:0'), I get pool3 with shape of [?, 2048], which makes the other line tf.matmul(tf.squeeze(pool3, [1, 2]), w) hard to understand. why do you need to squeeze pool3 ?

I'm sorry that I made a mistake.