YoungBigBird

gumbel_hard = tf.cast(tf.equal(gumbel_softmax, tf.reduce_max(gumbel_softmax, 1, keep_dims=True)), tf.float32) mask = tf.stop_gradient(gumbel_hard - gumbel_softmax) + gumbel_softmax For the above code, it seems to choose the maximal index marked with "1" and all...

Dear Authors, In the code: sampled_character_folders = random.sample(folders, self.num_classes) random.shuffle(sampled_character_folders) labels_and_images = get_images(sampled_character_folders, range(self.num_classes), nb_samples=self.num_samples_per_class, shuffle=False) One of random labels (1 to 5 class) are assigned to each folder of...