我看了官方的文档,就几行代码,有没有一个全的代码呀。。
官方的代码全是 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")
到底如何微调?