Fixes:GitHub Issue #39976
PiperOrigin-RevId: 314177901 Change-Id: I5ce76f1d5cabb2f70e24dd095ebe5d3c412cc016
This commit is contained in:
parent
1a430ba06b
commit
25213f58c4
@ -1066,8 +1066,8 @@ class RandomContrast(Layer):
|
|||||||
else:
|
else:
|
||||||
self.lower = self.upper = factor
|
self.lower = self.upper = factor
|
||||||
if self.lower < 0. or self.upper < 0. or self.lower > 1.:
|
if self.lower < 0. or self.upper < 0. or self.lower > 1.:
|
||||||
raise ValueError('Factor cannot have negative values, '
|
raise ValueError('Factor cannot have negative values or greater than 1.0,'
|
||||||
'got {}'.format(factor))
|
' got {}'.format(factor))
|
||||||
self.seed = seed
|
self.seed = seed
|
||||||
self.input_spec = InputSpec(ndim=4)
|
self.input_spec = InputSpec(ndim=4)
|
||||||
super(RandomContrast, self).__init__(name=name, **kwargs)
|
super(RandomContrast, self).__init__(name=name, **kwargs)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user