我正在使用这个命令:
tfjs.converters.save_keras_model(model,'jsmodels')
但是我得到了一个 model.json 和 3 个权重文件
group1-shard1of3.bin
group1-shard2of3.bin
group1-shard3of3.bin
我只想获取一个 .bin 文件，我该怎么做？
最佳答案
我不太确定使用 save_keras_model 是否可以做到这一点，但从命令行使用 tensorflowjs_converter 我会执行以下操作。您将 --weigth_shard_size_bytes 指定为您拥有的模型的大小。如果您的模型 <= 30Mb，则将其设置为 30000000 字节 将生成单个文件 group1-shard1of1.bin。
tensorflowjs_converter --input_format keras --weight_shard_size_bytes 30000000 'model.h5' 'output_dir'
关于tensorflow - 如何在tensorflow js转换器中获取单个权重文件，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/66619509/