fix torch test on mix_up (#20623)

keras-team · Dec 10, 2024 · a0f586b · a0f586b
1 parent 9294db1
commit a0f586b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras/src/layers/preprocessing/image_preprocessing/mix_up.py b/keras/src/layers/preprocessing/image_preprocessing/mix_up.py
@@ -68,6 +68,7 @@ def get_random_transformation(self, data, training=True, seed=None):
         }
 
     def transform_images(self, images, transformation=None, training=True):
+        images = self.backend.cast(images, self.compute_dtype)
         mix_weight = transformation["mix_weight"]
         permutation_order = transformation["permutation_order"]