search

xiaojiew94 / KDGAN

37 stars 11 forks source link

TensorFlow version for MNIST? #2

Open florinmatei opened 5 years ago

florinmatei commented 5 years ago

What TensorfFlow version is needed to run the MNIST example? Thank you!

florinmatei commented 5 years ago

I think the error below is due to TensorFlow version mismatch:

Traceback (most recent call last): File "pretrain_gen.py", line 28, in tn_gen = GEN(flags, mnist.train, is_training=True) File "/Users/florinmatei/Projects/KDGAN2/KDGAN/mdlcompr/gen_model.py", line 39, in init self.gumbel_labels = gumbel_softmax(self.logits, self.gumbel_temperature) File "/Users/florinmatei/Projects/KDGAN2/KDGAN/mdlcompr/gumbel_softmax.py", line 11, in gumbel_softmax y = logits + sample_gumbel(tf.shape(logits)) AttributeError: 'Session' object has no attribute 'shape'