将keras模型转换为react-native的tfjs时更改分片大小

Question

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

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

无论如何，是否有将我的分片组合成1个分片，或在转换时更改分片的大小，使我最终只能得到一个分片？

Answer 1

可以指定分片文件的大小。大小越大，文件数量越少。可以使用--weight_shard_size_bytes

的选项tfjs_converter指定分片文件的大小。