在Keras中，在模型中使用Lambda时无法保存模型检查点。给出错误ValueError:只能将大小为1的数组转换为Python标量

from keras.models import Model
from keras.layers import Input, Lambda, Flatten, Dense

# 定义输入层
input_layer = Input(shape=(32, 32, 3))

# 使用Lambda层进行自定义操作
lambda_layer = Lambda(lambda x: x + 1)(input_layer)

# 使用Flatten层将数组展平为标量值
flatten_layer = Flatten()(lambda_layer)

# 添加其他层和模型结构
# ...

# 定义模型
model = Model(inputs=input_layer, outputs=flatten_layer)

# 编译和训练模型
# ...

# 保存模型检查点
model.save_weights('model_checkpoint.h5')