Skip to content

Commit

Permalink
Correct bug for MixUp initialization. (#20630)
Browse files Browse the repository at this point in the history
* Correct bug for MixUp initialization.

* Update format indent
  • Loading branch information
shashaka authored Dec 11, 2024
1 parent e12ae46 commit 57e29a6
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ class MixUp(BaseImagePreprocessingLayer):
"""

def __init__(self, alpha=0.2, data_format=None, seed=None, **kwargs):
super().__init__(data_format=None, **kwargs)
super().__init__(data_format=data_format, **kwargs)
self.alpha = alpha
self.seed = seed
self.generator = SeedGenerator(seed)
Expand Down

0 comments on commit 57e29a6

Please sign in to comment.