python - 没有使用 Dist-Keras 更新权重

Question

正如这里所解释的，当我运行示例笔记本时，我的权重在训练后永远不会更新：

model.layers[0].get_weights()
trained_model.layers[0].get_weights()

两者都给我相同的权重（好像模型根本没有训练过）：

[array([[-0.39513412, 0.26937097, -0.36478603, 0.30427128, -0.13985097,
-0.22316453, 0.13130313, -0.08426034],
[ 0.41418487, -0.46847233, 0.58078319, -0.63027477, -0.45647684,
-0.325973 , 0.22211522, 0.55291325],
[ 0.54379755, -0.30091569, -0.02049094, -0.4734239 , -0.41363743,
-0.38102722, -0.19341171, -0.36358535],
[-0.08354402, 0.39400059, 0.04485017, -0.1212253 , 0.07950532,
0.37202805, 0.30843312, -0.25526762]], dtype=float32),
array([ 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32)]

我正在使用数据科学经验。有没有人让这个包在这个平台上工作？

Answer 1

我的问题是由于我的参数。我的数据集很小，所以我需要减小批量大小并增加 epoch。我希望这对其他人有帮助。