我已使用tensorflowjs_converter将Keras模型转换为TensorflowJS。它生成了一个'model.json'文件和9个'* .bin'文件。我正在尝试使用以下功能加载此模型。
const model = await tf.loadLayersModel(bundleResourceIO(modelJson, modelWeights));
modelWeights仅选择一个权重。但是我有9个人。如何正确加载此模型？
答案 0 :(得分：3)
好的，我刚刚找到了解决方案。基本上tensorflow_converter将权重分成一些较小的大小。但是，如果增加重量文件的大小限制，则生成的文件数将减少。假设您的模型权重文件占用30MB。因此，将限制增加到> = 30MB，它将仅生成单个权重文件。您可以按照以下命令进行操作：
tensorflowjs_converter --input_format keras --weight_shard_size_bytes 60000000 'your_model.h5' 'output_dir'