将 keras 模型转换为 tfjs 以进行反应本机时更改分片大小

Change shard size when converting keras model to tfjs for react-native

我正在尝试将 keras 模型转换为 tensorflowjs,以便它可以 运行 与 react-native。

React-native 似乎只支持有 1 个分片的 tfjs 模型。

是否可以将我的分片合并为 1 个分片,或者在转换时更改分片大小以便我最终只得到一个分片?

可以指定分片文件大小。大小越大,文件数量就越少。可以使用 tfjs_converter

的选项 --weight_shard_size_bytes 指定分片文件大小