为 tensorflow.js 模型实现同步 fit() 或训练函数
[英]Implement synchronous fit() or training function for tensorflow.js model
问题描述
我有一个使用 tensorflow.js 神经网络玩“井字游戏”的代理，试图模仿 Q 函数的行为。
我正在关注此旅行报告，但使用 TFJS 和节点： 第 4 部分 — 神经网络 Q 学习
代理在游戏过程中跟踪游戏状态，并在游戏完成后尝试根据游戏结束方式使用这些状态的更新 Q 值来训练神经网络。
我的问题是：fit() 是一个异步函数，我玩游戏的脚本运行一个游戏循环，游戏循环完成然后我得到一堆 UnhandledPromiseRejectionWarning 因为之前所有游戏都调用了 fit()正在进行中。
有没有办法在每场比赛结束时同步训练我的模型。 也许使用 model.makeTrainFunction() 或者有没有办法使用优化器？ （在购物车杆示例中找到了 optimizer.applyGradients() ）
链接到我的代码NNQPlayer.js
感谢您阅读我的问题。 我希望有人能在正确的方向上给我一个推动！
1 个解决方案
解决方案1
1 已采纳 2019-03-21 15:30:55
fit是一个异步操作。 如果您正在寻找一种多次调用fit的方法，您只需在调用fit之前使用await 。 这将确保fit操作在另一个调用fit之前完成。 你可以看看这个答案
如何修复“model.predict 不是函数”（tensorflow.js）？
[英]How to fix “model.predict is not a function” (tensorflow.js)?
如何使用Node.js（tfjs-node）从Tensorflow.js中的检查点重新启动模型训练？
[英]How to restart a model training from a check point in Tensorflow.js with Node.js (tfjs-node)?
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.
相关问题
使用 tensorflow.js 在 model.fit 中详细说明
通用 Tensorflow.js 训练示例
找到TensorFlow.js训练进度
如何修复“model.predict 不是函数”（tensorflow.js）？
Tensorflow.js加载模型返回函数预测未定义
如何使用Node.js（tfjs-node）从Tensorflow.js中的检查点重新启动模型训练？
tensorflow.js模型无法正确预测
Tensorflow.js Model 尺寸不匹配
在本地将 keras model 加载到 tensorflow.js 中
如何更新Tensorflow.js模型的权重？
相关标签