我应该为tensorflow使用哪个模型(contrib或模型)?

时间:2019-01-25 02:56:31

标签: tensorflow deep-learning resnet tensorflow-model-analysis

例如,如果我想使用renset_v2,则在tensorflow上有两个模型文件:

一个是here,另一个是heremodels/researchtensorflow/contrib中都有很多张量流模型。

我很困惑:哪种型号更好?我应该使用哪种型号?

2 个答案:

答案 0 :(得分:2)

通常,tf.contrib包含主要由社区提供的代码。它旨在包含最终应合并到核心TensorFlow中的功能和贡献,但其接口可能仍会更改,或者需要进行一些测试以查看它们是否可以得到更广泛的接受。

Tensorflow团队不支持tf.contrib中的代码。希望对它有所帮助,但它可能随时更改或删除;没有保证。

tf.research文件夹包含研究人员在TensorFlow中实现的机器学习模型。这些模型由其各自的作者维护,因此被淘汰的可能性较低。

另一方面,直接提供的模型得到Tensorflow团队的正式支持,并且通常被首选,因为它们在将来的发行版中被淘汰的可能性较低。如果您同时实现了模型,则通常应避免使用contrib版本考虑到将来的兼容性,但是社区确实在其中做了一些很棒的工作,因此您可能会发现主存储库中不存在一些模型/工作,但是如果直接从contrib分支中使用它们会有所帮助。

还要注意短语通常避免,因为它与应用程序有关。

希望能回答您的问题,并提出疑问。

答案 1 :(得分:1)

使用Tensorflow 2.0(即将推出)tf.contrib将被删除。

因此,如果您希望项目是最新的并且在接下来的几个月中仍可以工作,则必须开始使用models/research