search

igul222 / improved_wgan_training

Code for reproducing experiments in "Improved Training of Wasserstein GANs"
MIT License
2.35k stars 668 forks source link

inception_score.py: ValueError in the method _init_inception() #82

Closed psanch21 closed 5 years ago

psanch21 commented 5 years ago

I am getting the following error in the method _init_inception() after fixing o.set_shape(tf.TensorShape(new_shape)) The error is ValueError: Shape must be rank 2 but is rank 1 for 'MatMul' (op: 'MatMul') with input shapes: [2048], [2048,1008].

which is in line 93 logits = tf.matmul(tf.squeeze(pool3), w)

Oreobird commented 5 years ago

logits= tf.matmul(tf.squeeze(pool3, [1, 2]), w) works for me