时间戳

首页 > 解决方案 > 如何在 tensorflow js 转换器中获取单个权重文件

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

问题描述

我正在使用这个命令：

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

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

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

我只想得到一个 .bin 文件，我该怎么做？

标签： tensorflowtensorflow.jstensorflowjs-converter

解决方案

我不太确定这是否可以使用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'

推荐阅读