将keras模型转换为重量不匹配的tensorflowjs,层组=2

将keras模型转换为重量不匹配的tensorflowjs,层组=2,keras,tensorflow.js,Keras,Tensorflow.js,我正在构建一个keras模型,将预先设定的权重加载到模型中,然后尝试将其转换为tfjs,问题出现在将groups参数设置为2的某一层。我将权重形状与预渲染形状和keras模型进行了比较,结果是匹配的,但是当我将其转换为tfjs并加载到brower上时,它给出了一个不匹配的错误 代码: 正确的权重应该是:[5,5,48,256],这是预训练权重的形状,但当我将其转换并加载到brower时,错误发生了: Uncaught (in promise) Error: Shape mismatch: [5,

我正在构建一个keras模型,将预先设定的权重加载到模型中,然后尝试将其转换为tfjs,问题出现在将groups参数设置为2的某一层。我将权重形状与预渲染形状和keras模型进行了比较,结果是匹配的,但是当我将其转换为tfjs并加载到brower上时,它给出了一个不匹配的错误

代码:

正确的权重应该是:[5,5,48,256],这是预训练权重的形状,但当我将其转换并加载到brower时,错误发生了:

Uncaught (in promise) Error: Shape mismatch: [5,5,96,256] vs. [5,5,48,256]
at checkShapesMatch (webgazer.js:66610)
at variables_LayerVariable.write (webgazer.js:66578)
at webgazer.js:66781
at Array.forEach ()
at batchSetValue (webgazer.js:66779)
at models_Sequential.loadWeights (webgazer.js:70579)
at loadLayersModelFromIOHandler (webgazer.js:73987)
转换通过以下方式完成: tfjs.converts.save_keras_model(model)

Uncaught (in promise) Error: Shape mismatch: [5,5,96,256] vs. [5,5,48,256]
at checkShapesMatch (webgazer.js:66610)
at variables_LayerVariable.write (webgazer.js:66578)
at webgazer.js:66781
at Array.forEach ()
at batchSetValue (webgazer.js:66779)
at models_Sequential.loadWeights (webgazer.js:70579)
at loadLayersModelFromIOHandler (webgazer.js:73987)