我将在Keras训练的模型保存为.h5文件。 当尝试在分布式设置中加载它时，我收到错误：
InvalidArgumentError（请参阅上面的回溯）：无法分配设备 节点'policy / dense_2_b'：无法满足显式设备 规范'/ job：ps / task：0'因为没有匹配的设备 规范在此过程中注册;可用设备： /作业：本地主机/复制：0 /任务：0 / CPU：0
似乎.h5文件中的某些变量被分配给具有作业“localhost”的特定设备。因此，当我尝试在参数服务器上加载它时，我收到错误。
有人可以澄清如何解决这个问题吗？我可能应首先加载keras模型而不启动服务器，然后在参数服务器上重新加载它。但细节对我来说还不清楚..