sudo-paras-shah commited on
Commit
377fcf0
·
1 Parent(s): b78bb63

Finally fix mobilenet error

Browse files
Files changed (1) hide show
  1. src/nets/mobilenet.py +1 -1
src/nets/mobilenet.py CHANGED
@@ -98,7 +98,7 @@ def MobileNet(input_shape=None,
98
  return model
99
 
100
  def relu6(x):
101
- return ReLU(x, max_value=6)
102
 
103
  if __name__ == '__main__':
104
  model = MobileNet(input_shape=(224, 224, 3))
 
98
  return model
99
 
100
  def relu6(x):
101
+ return ReLU(max_value=6)(x)
102
 
103
  if __name__ == '__main__':
104
  model = MobileNet(input_shape=(224, 224, 3))