Skip to content

Commit

Permalink
Fix YAML serialization for Advanced Activations
Browse files Browse the repository at this point in the history
Fix keras-team#2871 (comment)

The problem occurs because PyYAML can't recognize numpy's data types.
  • Loading branch information
henry0312 committed Aug 4, 2016
1 parent 0588393 commit 745d353
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions keras/layers/advanced_activations.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ def call(self, x, mask=None):
return pos + self.alpha * (K.exp(neg) - 1.)

def get_config(self):
config = {'alpha': self.alpha}
config = {'alpha': float(self.alpha)}
base_config = super(ELU, self).get_config()
return dict(list(base_config.items()) + list(config.items()))

Expand Down Expand Up @@ -161,8 +161,8 @@ def call(self, x, mask=None):
return K.softplus(self.betas * x) * self.alphas

def get_config(self):
config = {'alpha_init': self.alpha_init,
'beta_init': self.beta_init}
config = {'alpha_init': float(self.alpha_init),
'beta_init': float(self.beta_init)}
base_config = super(ParametricSoftplus, self).get_config()
return dict(list(base_config.items()) + list(config.items()))

Expand Down Expand Up @@ -195,7 +195,7 @@ def call(self, x, mask=None):
return x * K.cast(x > self.theta, K.floatx())

def get_config(self):
config = {'theta': self.theta}
config = {'theta': float(self.theta)}
base_config = super(ThresholdedReLU, self).get_config()
return dict(list(base_config.items()) + list(config.items()))

Expand Down

0 comments on commit 745d353

Please sign in to comment.