tfhub调用本地模型retrain

最新推荐文章于 2020-08-17 16:58:27 发布

shengchun75

最新推荐文章于 2020-08-17 16:58:27 发布

阅读量1k

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/shengchun75/article/details/97474183

版权

深度学习专栏收录该内容

6 篇文章 0 订阅

订阅专栏

参考知乎文章tensorflow_hub迁移学习(分类demo)
retrain.py文件中可以自动下载预训练模型，但是在国内下不来，国外也有人反馈说下不来。解决方案是搭梯子用下载工具来下，解压后使用本地模型训练。
具体方法是：
对于Google提供的–tfhub_module模型参数，https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1，在后面加上?tf-hub-format=compressed变成https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1?tf-hub-format=compressed即可下载模型到本地