add missing config line

This commit is contained in:
Michael Oliver
2017-04-21 17:57:00 -07:00
parent 1b11a5d3dd
commit e6ef35ec70
+2 -1
View File
@@ -229,7 +229,8 @@ class SReLU(Layer):
't_left_initializer': self.t_left_initializer,
'a_left_initializer': self.a_left_initializer,
't_right_initializer': self.t_right_initializer,
'a_right_initializer': self.a_right_initializer
'a_right_initializer': self.a_right_initializer,
'shared_axes': self.shared_axes
}
base_config = super(SReLU, self).get_config()
return dict(list(base_config.items()) + list(config.items()))