背景:使用的模型需要转换成tflite,用tf.keras重写后,因为用到了 tf.keras.activations operators 所以在model.save的时候会报错Not JSON Serializable error。应该是类型不支持,造成node json序列化失败。
解决方法:
1. 官方tf nightly build '2.0.0-dev20190802' 版本已经fixed 这个bug.
2. 自己修改源码。修改方法
3.
tf.keras.layers.Lambd(lambda t: tf.keras.activations.sigmoid(t))(x)