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 35, in init
self.gumbel_labels = utils.gumbel_softmax(self.logits, self.gumbel_temperature)
AttributeError: 'module' object has no attribute 'gumbel_softmax'
There is a gumbel_softmax.py in mdlcompr folder but if I'm trying to use it directly (by importing it in get_model.py module) I'm running into issues with TensorFlow saying that tf.shape is not a valid method so knowing what TensorFlow version is another important thing we need to know.
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 35, in init
self.gumbel_labels = utils.gumbel_softmax(self.logits, self.gumbel_temperature)
AttributeError: 'module' object has no attribute 'gumbel_softmax'
There is a gumbel_softmax.py in mdlcompr folder but if I'm trying to use it directly (by importing it in get_model.py module) I'm running into issues with TensorFlow saying that tf.shape is not a valid method so knowing what TensorFlow version is another important thing we need to know.