tfhub调用本地模型retrain_tf-hub使用本地模型

作者：我家自动化 | 2024-05-03 23:16:43

踩

tf-hub使用本地模型

参考知乎文章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即可下载模型到本地

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家自动化/article/detail/532032