首页 > 解决方案 > 如何限制 Keras 中权重的格式

问题描述

在 Keras(张量流后端)中,神经网络的权重和偏差默认为 float32。有没有办法改变它并提供不同的格式?

提前致谢。

标签: pythontensorflowmachine-learningneural-networkkeras

解决方案


它定义在~/.keras/keras.json

{
    "image_data_format": "channels_last",
    "floatx": "float32",
    "backend": "tensorflow",
    "epsilon": 1e-07
}

您可以更改floatx以更改默认浮点格式。从文档中:

floatx:字符串、“float16”、“float32”或“float64”。默认浮点精度。


推荐阅读