Skip to content

Commit

Permalink
syntax fix (#898)
Browse files Browse the repository at this point in the history
  • Loading branch information
anas-rz authored Sep 16, 2023
1 parent 7486af9 commit 74a4e7f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/keras_io/vision/cct.py
Original file line number Diff line number Diff line change
Expand Up @@ -256,7 +256,7 @@ def call(self, x, training=None):
if training:
keep_prob = 1 - self.drop_prob
shape = (keras.ops.shape(x)[0],) + (1,) * (len(x.shape) - 1)
random_tensor = keep_prob + keras.random.random.uniform(shape, 0, 1)
random_tensor = keep_prob + keras.random.uniform(shape, 0, 1)
random_tensor = keras.ops.floor(random_tensor)
return (x / keep_prob) * random_tensor
return x
Expand Down

0 comments on commit 74a4e7f

Please sign in to comment.