Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
elad-c committed Sep 22, 2024
1 parent b37f675 commit ad06ecc
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def create_networks(self):
x = tf.concat([x1, x2, as_const(x3), x3], 1)

inds = tf.reshape(tf.argmax(tf.reshape(x, (-1, 192 * 32, 38)), axis=1), (-1, 1, 1, 38))
b = tf.gather(np.random.random((100,)).astype(np.float32), inds)
b = tf.gather(np.random.random((192 * 32,)).astype(np.float32), inds)
x = tf.add(x, b)

return tf.keras.models.Model(inputs=inputs, outputs=x)
Expand Down

0 comments on commit ad06ecc

Please sign in to comment.