javascript-如何在TensorFlow.js中的maxPooling层上使用optimizer.minimize

前端之家收集整理的这篇文章主要介绍了javascript-如何在TensorFlow.js中的maxPooling层上使用optimizer.minimize 前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

当在模型.predict()中使用optimizer.minimize()来训练TensorFlow.js中具有损失函数的模型时,我遇到了一个问题.仅当我在卷积神经网络中使用maxPooling2D层且代码类似于以下代码时,才会发生这种情况.产生此错误:无法读取未定义的属性“后端”.我不确定是什么原因引起的或如何解决.当使用没有任何池化层的卷积层(tf.layers.conv2d())时,不会发生此错误.我正在使用TensorFlow.js版本0.14.2和Google Chrome版本71.0.3578.98.可以使用以下代码来重现此错误

  1. loss = (pred,label) => pred.sub(label).square().mean();
  2. optimizer = tf.train.sgd(0.001);
  3. const input = tf.input({shape: [100,100,4]});
  4. const conv = tf.layers.conv2d({
  5. kernelSize: 5,filters: 8,strides: 1,activation: 'relu',kernelInitializer: 'VarianceScaling'
  6. });
  7. const pool = tf.layers.maxPooling2d({
  8. poolSize: [2,2],strides: [2,2]
  9. });
  10. const flat = tf.layers.flatten();
  11. const dense = tf.layers.dense({units: 10});
  12. const output = dense.apply(flat.apply(pool.apply(conv.apply(input))));
  13. const model = tf.model({inputs: input,outputs: output});
  14. for (var i = 0; i < 10; i++) {
  15. optimizer.minimize(() =>
  16. loss(model.predict([tf.ones([1,4])]),tf.ones([1,10]))
  17. );
  18. }

编辑:这已经解决.有关详细信息,请参见scai’s answer.

编辑2:这似乎不是错误,而是对model.predict()的反向传播进行了更改. More information

最佳答案
在TensorFlow.js 0.14中,有一个更改禁用了Model.predict()方法中的反向传播支持.您可以将Model.apply()方法与{training:true}标志一起使用来修复代码.

即改变

  1. optimizer.minimize(() =>
  2. loss(model.predict([tf.ones([1,10]))
  3. );

  1. optimizer.minimize(() =>
  2. loss(model.apply([tf.ones([1,4])],{training: true}),10]))
  3. );

猜你在找的JavaScript相关文章