如何在tensorflow js转换器中获取单个权重文件

Question

我正在使用这个命令：

tfjs.converters.save_keras_model(model,'jsmodels')

但我得到一个 model.json 和 3 个权重文件

group1-shard1of3.bin
group1-shard2of3.bin
group1-shard3of3.bin

我只想获取一个 .bin 文件，我该怎么做？

Answer 1

我不太确定这是否可以使用 save_keras_model 但从命令行使用 tensorflowjs_converter 我会执行以下操作。您将 --weigth_shard_size_bytes 指定为您拥有的模型的大小。如果您的模型 <= 30Mb，则将其设置为 30000000 bytes 将导致单个文件 group1-shard1of1.bin.

tensorflowjs_converter --input_format keras --weight_shard_size_bytes 30000000 'model.h5' 'output_dir'

如何在tensorflow js转换器中获取单个权重文件

how to get a single weights file in tensorflow js converter

tensorflow

tensorflow.js

tensorflowjs-converter