Tensorflow 1.x 版本如何对 Tensorflow hub 的模型进行微调?

2021-02-04 17:25:25 +08:00
 waitfor

我看了官方的文档,就几行代码,有没有一个全的代码呀。。 官方的代码全是 TF2 keras 的,难受! 急急急! https://hub.tensorflow.google.cn/google/imagenet/resnet_v2_50/classification/3以这个为例,就只有几行代码,

module = hub.Module("https://hub.tensorflow.google.cn/google/imagenet/resnet_v2_50/classification/3",
                    trainable=True, tags={"train"})
logits = module(inputs=dict(images=images, batch_norm_momentum=0.997),
                signature="image_classification_with_bn_hparams")

到底如何微调?

1932 次点击
所在节点    TensorFlow
0 条回复

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/751322

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX