Tensorflow.js图层模型和Graph模型有什么区别?

时间:2019-12-04 18:35:26

标签: python tensorflow deep-learning tensorflow.js

想知道thisthis有什么区别? 仅仅是输入变化的方式吗?

2 个答案:

答案 0 :(得分:1)

两者都在执行相同的任务,即将NN模型转换为tfjs格式。只是在第一个链接模型中使用了h5格式(通常是保存keras模型的格式),而在另一个链接模型中则是TF保存的模型。

答案 1 :(得分:1)

LayersModel和GraphModels之间的主要区别是:

  1. LayersModel只能从tf.keras或keras HDF5格式模型类型导入。可以从上述模型类型或TensorFlow SavedModels中导入GraphModels。

  2. LayersModels支持JavaScript的进一步培训(通过其fit()方法)。 GraphModel仅支持推断。

  3. GraphModel由于具有图形优化功能,因此通常比LayersModel具有更高的推理速度(10-20%),这归功于仅推理功能。

希望这会有所帮助。