dd2*_*210 5 distributed keras tensorflow
我将在 Keras 中训练的模型保存为 .h5 文件。当尝试在分布式设置中加载它时，我收到错误：
InvalidArgumentError（请参阅上面的回溯）：无法将设备分配给节点“policy/dense_2_b”：无法满足显式设备规范“/job:ps/task:0”，因为在此过程中没有注册与该规范匹配的设备；可用设备：/job:localhost/replica:0/task:0/cpu:0
似乎 .h5 文件中的变量以某种方式分配给具有作业“localhost”的特定设备。因此，当我尝试将其加载到参数服务器上时，我收到错误。
有人能澄清如何解决这个问题吗？我可能应该首先加载 keras 模型而不启动服务器，然后在参数服务器上重新加载它。但细节我不清楚..
| 归档时间： |
|
| 查看次数： |
724 次 |
| 最近记录： |