输入层中没有任何形状会导致对从TF.keras Python转换的TensroFlow.js模型的错误预测

时间:2020-05-31 05:21:23

标签: tensorflow tensorflow.js

我正在草绘rnn,在tf.keras环境中进行培训,并使用tfjs-converter将tf模型转换为tfjs版本。但是我们的tfjs环境的结果远远超出了我们的预期。我们已经通过更改输入形状来解决此性能问题

decoder_input = Input(shape=(None, 5), name='decoder_input')

decoder_input = Input(shape=(1, 5), name='decoder_input')

但是问题是我们不知道为什么这种差异会导致性能下降。在我们的原始代码中,输入形状不会对我们的predict.js造成任何错误,只是性能很差。有人也遇到过这样的问题吗?还是有人知道这里发生了什么?非常感谢!

0 个答案:

没有答案